/external/tensorflow/tensorflow/core/distributed_runtime/ |
H A D | rpcbench_test.cc | 124 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 D | message_wrappers.cc | 781 return response_.tensor_size(); 833 return response_->tensor_size();
|
/external/tensorflow/tensorflow/core/grappler/inputs/ |
H A D | trivial_test_graph_input_yielder.h | 31 TrivialTestGraphInputYielder(int num_stages, int width, int tensor_size,
|
H A D | trivial_test_graph_input_yielder.cc | 31 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 D | debug_fibonacci.py | 37 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 D | examples_test.sh | 62 cat << EOF | ${DEBUG_FIBONACCI_BIN} --tensor_size=2 --ui_type=readline
|
/external/tensorflow/tensorflow/contrib/mpi_collectives/ |
H A D | mpi_ops_test.py | 271 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 D | ndarray_tensor.cc | 216 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 D | tflite_driver.cc | 91 int tensor_size = tensor.bytes / sizeof(T); local 94 for (int i = 0; i < tensor_size; ++i) {
|
/external/tensorflow/tensorflow/core/kernels/ |
H A D | stack_ops.cc | 176 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 D | tensor_array_ops.cc | 183 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 D | tensor_slice_writer.cc | 121 << sts_.meta().tensor_size() << " tensors (" << file_size
|
H A D | tensor_slice_writer.h | 128 index = sts_.meta().tensor_size();
|
H A D | tensor_slice_writer_test.cc | 156 EXPECT_EQ(4, sts.meta().tensor_size());
|
/external/tensorflow/tensorflow/contrib/mpi/ |
H A D | mpi_rendezvous_mgr.cc | 182 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 D | linalg_ops.py | 278 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 D | virtual_scheduler.cc | 973 const auto tensor_size = local 975 total_output_size += tensor_size; 977 tensor_size); 979 tensor_size);
|
H A D | op_level_cost_estimator.cc | 828 int64 tensor_size = 1; local 833 tensor_size *= dim.size(); 835 return tensor_size;
|
/external/tensorflow/tensorflow/core/framework/ |
H A D | attr_value_util.cc | 115 } else if (attr_value.list().tensor_size() > 0) { 116 for (int i = 0; i < attr_value.list().tensor_size(); ++i) {
|
H A D | op_def_util.cc | 114 length = attr_value.list().tensor_size();
|
/external/tensorflow/tensorflow/contrib/verbs/ |
H A D | rdma.cc | 1532 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 D | cc_op_gen.cc | 258 } 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 D | tensor_bundle_test.cc | 828 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 D | python_op_gen.cc | 385 } else if (value.list().tensor_size() > 0) { 386 for (int i = 0; i < value.list().tensor_size(); ++i) {
|
/external/tensorflow/tensorflow/c/ |
H A D | c_api_test.cc | 2145 const size_t tensor_size[] = {TF_ARRAYSIZE(tensor1), TF_ARRAYSIZE(tensor2)}; local 2157 EXPECT_EQ(sizeof(char) * tensor_size[i], TF_TensorByteSize(v)) << i;
|