aliran tensor:: operasi:: Penghalang

#include <data_flow_ops.h>

Mendefinisikan penghalang yang bertahan di berbagai eksekusi grafik.

Ringkasan

Penghalang mewakili peta nilai kunci, di mana setiap kunci adalah string, dan setiap nilai adalah tupel tensor.

Pada saat runtime, penghalang berisi elemen 'lengkap' dan 'tidak lengkap'. Elemen lengkap telah menentukan tensor untuk semua komponen tupel nilainya, dan dapat diakses menggunakan BarrierTakeMany . Elemen yang tidak lengkap memiliki beberapa komponen yang tidak terdefinisi dalam tupel nilainya, dan dapat diperbarui menggunakan BarrierInsertMany .

Argumen:

  • ruang lingkup: Objek Lingkup
  • komponen_tipe: Jenis setiap komponen dalam suatu nilai.

Atribut opsional (lihat Attrs ):

  • bentuk : Bentuk setiap komponen dalam suatu nilai. Setiap bentuk harus bernilai 1 pada dimensi pertama. Panjang attr ini harus sama dengan panjang komponen_tipe.
  • kapasitas: Kapasitas penghalang. Kapasitas defaultnya adalah MAX_INT32, yang merupakan kapasitas terbesar dari antrean yang mendasarinya.
  • wadah: Jika tidak kosong, pembatas ini ditempatkan pada wadah yang diberikan. Jika tidak, wadah default akan digunakan.
  • shared_name: Jika tidak kosong, penghalang ini akan dibagikan dengan nama tertentu di beberapa sesi.

Pengembalian:

  • Output : Pegangan untuk penghalang.

Konstruktor dan Destruktor

Barrier (const :: tensorflow::Scope & scope, const DataTypeSlice & component_types)
Barrier (const :: tensorflow::Scope & scope, const DataTypeSlice & component_types, const Barrier::Attrs & attrs)

Atribut publik

handle
operation

Fungsi publik

node () const
::tensorflow::Node *
operator::tensorflow::Input () const
operator::tensorflow::Output () const

Fungsi statis publik

Capacity (int64 x)
Container (StringPiece x)
Shapes (const gtl::ArraySlice< PartialTensorShape > & x)
SharedName (StringPiece x)

Struktur

tensorflow:: ops:: Penghalang:: Attrs

Penyetel atribut opsional untuk Barrier .

Atribut publik

menangani

::tensorflow::Output handle

operasi

Operation operation

Fungsi publik

Penghalang

 Barrier(
  const ::tensorflow::Scope & scope,
  const DataTypeSlice & component_types
)

Penghalang

 Barrier(
  const ::tensorflow::Scope & scope,
  const DataTypeSlice & component_types,
  const Barrier::Attrs & attrs
)

simpul

::tensorflow::Node * node() const 

operator::tensorflow::Masukan

 operator::tensorflow::Input() const 

operator::tensorflow::Keluaran

 operator::tensorflow::Output() const 

Fungsi statis publik

Kapasitas

Attrs Capacity(
  int64 x
)

Wadah

Attrs Container(
  StringPiece x
)

Bentuk

Attrs Shapes(
  const gtl::ArraySlice< PartialTensorShape > & x
)

Nama Bersama

Attrs SharedName(
  StringPiece x
)