Searched refs:tensor_size (Results 1 - 25 of 25) sorted by relevance

/external/tensorflow/tensorflow/core/distributed_runtime/
H A Drpcbench_test.cc124 GraphDef CreateGraphDef(int num_stages, int width, int tensor_size, argument
133 Output x = Const(s.WithOpName("x"), 0.0f, {tensor_size, 1});
157 string DebugString(const Tensor& x, const Tensor& y, int tensor_size) { argument
158 CHECK_EQ(x.NumElements(), tensor_size);
159 CHECK_EQ(y.NumElements(), tensor_size);
163 CHECK_GE(tensor_size, 2);
169 static void BM_Helper(int iters, int width, int num_stages, int tensor_size, argument
176 GraphDef def = CreateGraphDef(num_stages, width, tensor_size,
183 Tensor x(DT_FLOAT, TensorShape({tensor_size, 1}));
188 "; tensor bytes/send: ", tensor_size * sizeo
235 BM_RPC(int iters, int width, int tensor_size) argument
[all...]
H A Dmessage_wrappers.cc781 return response_.tensor_size();
833 return response_->tensor_size();
/external/tensorflow/tensorflow/core/grappler/inputs/
H A Dtrivial_test_graph_input_yielder.h31 TrivialTestGraphInputYielder(int num_stages, int width, int tensor_size,
H A Dtrivial_test_graph_input_yielder.cc31 GraphDef CreateGraphDef(int num_stages, int width, int tensor_size, argument
39 const int batch_size = tensor_size < 0 ? 1 : tensor_size;
90 int num_stages, int width, int tensor_size, bool insert_queue,
94 tensor_size_(tensor_size),
/external/tensorflow/tensorflow/python/debug/examples/
H A Ddebug_fibonacci.py37 np.ones([FLAGS.tensor_size] * 2), dtype=tf.int32, name="node_00")
39 np.ones([FLAGS.tensor_size] * 2), dtype=tf.int32, name="node_01")
71 "--tensor_size",
H A Dexamples_test.sh62 cat << EOF | ${DEBUG_FIBONACCI_BIN} --tensor_size=2 --ui_type=readline
/external/tensorflow/tensorflow/contrib/mpi_collectives/
H A Dmpi_ops_test.py271 tensor_size = [17] * 3
272 tensor_size[1] = 10 * (rank + 1)
273 tensor = tf.ones(tensor_size, dtype=tf.float32) * rank
288 tensor_size = [17] * 3
290 tensor = tf.ones(tensor_size, dtype=dtype) * rank
/external/tensorflow/tensorflow/python/lib/core/
H A Dndarray_tensor.cc216 const size_t tensor_size = TF_TensorByteSize(src); local
217 const char* limit = static_cast<const char*>(tensor_data) + tensor_size;
227 if (expected_tensor_size - tensor_size) {
231 " strings, but the tensor is encoded in ", tensor_size, " bytes");
/external/tensorflow/tensorflow/contrib/lite/testing/
H A Dtflite_driver.cc91 int tensor_size = tensor.bytes / sizeof(T); local
94 for (int i = 0; i < tensor_size; ++i) {
/external/tensorflow/tensorflow/core/kernels/
H A Dstack_ops.cc176 const Tensor* tensor_size; variable
177 OP_REQUIRES_OK(ctx, ctx->input("max_size", &tensor_size));
179 OP_REQUIRES(ctx, TensorShapeUtils::IsScalar(tensor_size->shape()),
182 tensor_size->shape().DebugString()));
184 int32 size_value = tensor_size->scalar<int32>()();
H A Dtensor_array_ops.cc183 const Tensor* tensor_size; variable
184 TF_RETURN_IF_ERROR(ctx->input("size", &tensor_size));
186 if (!TensorShapeUtils::IsScalar(tensor_size->shape())) {
189 tensor_size->shape().DebugString());
191 const int32 size = tensor_size->scalar<int32>()();
/external/tensorflow/tensorflow/core/util/
H A Dtensor_slice_writer.cc121 << sts_.meta().tensor_size() << " tensors (" << file_size
H A Dtensor_slice_writer.h128 index = sts_.meta().tensor_size();
H A Dtensor_slice_writer_test.cc156 EXPECT_EQ(4, sts.meta().tensor_size());
/external/tensorflow/tensorflow/contrib/mpi/
H A Dmpi_rendezvous_mgr.cc182 const int tensor_size = static_cast<int>(val.TotalBytes());
190 MPI_CHECK(MPI_Isend(temp, tensor_size, MPI_CHAR, mpi_dst, TAG_SENDTENSOR2,
/external/tensorflow/tensorflow/python/ops/
H A Dlinalg_ops.py278 tensor_size = tensor_shape.num_elements() * matrix.dtype.size
281 if tensor_size > L2_CACHE_SIZE_GUESSTIMATE and is_io_bound:
/external/tensorflow/tensorflow/core/grappler/costs/
H A Dvirtual_scheduler.cc973 const auto tensor_size = local
975 total_output_size += tensor_size;
977 tensor_size);
979 tensor_size);
H A Dop_level_cost_estimator.cc828 int64 tensor_size = 1; local
833 tensor_size *= dim.size();
835 return tensor_size;
/external/tensorflow/tensorflow/core/framework/
H A Dattr_value_util.cc115 } else if (attr_value.list().tensor_size() > 0) {
116 for (int i = 0; i < attr_value.list().tensor_size(); ++i) {
H A Dop_def_util.cc114 length = attr_value.list().tensor_size();
/external/tensorflow/tensorflow/contrib/verbs/
H A Drdma.cc1532 size_t tensor_size = result_tensor_->TotalBytes();
1535 if (tensor_size == 0) {
1539 mr_ = RdmaMemoryMgr::Singleton().FindMemoryRegion(rdma_addr_, tensor_size);
1548 RdmaMemoryMgr::Singleton().FindMemoryRegion(rdma_addr_, tensor_size);
/external/tensorflow/tensorflow/cc/framework/
H A Dcc_op_gen.cc258 } else if (attr_value.list().tensor_size() > 0) {
259 for (int i = 0; i < attr_value.list().tensor_size(); ++i) {
/external/tensorflow/tensorflow/core/util/tensor_bundle/
H A Dtensor_bundle_test.cc828 int tensor_size) {
835 TF_CHECK_OK(writer.Add("big", Constant(32.1, TensorShape({tensor_size}))));
827 BM_BundleAlignmentByteOff(int iters, int alignment, int tensor_size) argument
/external/tensorflow/tensorflow/python/framework/
H A Dpython_op_gen.cc385 } else if (value.list().tensor_size() > 0) {
386 for (int i = 0; i < value.list().tensor_size(); ++i) {
/external/tensorflow/tensorflow/c/
H A Dc_api_test.cc2145 const size_t tensor_size[] = {TF_ARRAYSIZE(tensor1), TF_ARRAYSIZE(tensor2)}; local
2157 EXPECT_EQ(sizeof(char) * tensor_size[i], TF_TensorByteSize(v)) << i;

Completed in 624 milliseconds