| /external/tensorflow/tensorflow/lite/delegates/hexagon/builders/ |
| D | min_max_builder.cc | 29 const auto& b_tensor = context->tensors[b_tensor_id]; in PopulateSubGraph() local
35 TF_LITE_ENSURE_STATUS(ComputeAndAddMinAndMax(context, b_tensor)); in PopulateSubGraph()
|
| /external/tensorflow/tensorflow/core/kernels/rnn/ |
| D | lstm_ops.cc | 328 const Tensor* b_tensor = nullptr; in Compute() local
329 OP_REQUIRES_OK(ctx, ctx->input("b", &b_tensor)); in Compute()
363 OP_REQUIRES(ctx, b_tensor->dim_size(0) == cell_size * 4, in Compute()
365 "b.dim_size(0) != cell_size * 4: ", b_tensor->dim_size(0), in Compute()
424 wcf_tensor->vec<T>(), wco_tensor->vec<T>(), b_tensor->vec<T>(), in Compute()
486 const Tensor* b_tensor = nullptr; in Compute() local
487 OP_REQUIRES_OK(ctx, ctx->input("b", &b_tensor)); in Compute()
545 OP_REQUIRES(ctx, b_tensor->dim_size(0) == cell_size * 4, in Compute()
547 "b.dim_size(0) != cell_size * 4: ", b_tensor->dim_size(0), in Compute()
686 wco_tensor->vec<T>(), b_tensor->vec<T>(), i_tensor->matrix<T>(), in Compute()
[all …]
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | quantize_nodes_test.cc | 217 Tensor b_tensor(DT_FLOAT, TensorShape({k, n})); in TestQuantizeMatMul() local
218 test::FillValues<float>(&b_tensor, b_values); in TestQuantizeMatMul()
219 Output b_op = Const(root.WithOpName("b_op"), Input::Initializer(b_tensor)); in TestQuantizeMatMul()
380 Tensor b_tensor(DT_FLOAT, TensorShape({2, 2, 3})); in TestQuantizeConcat() local
381 test::FillValues<float>(&b_tensor, in TestQuantizeConcat()
383 Output b_op = Const(root.WithOpName("b_op"), Input::Initializer(b_tensor)); in TestQuantizeConcat()
1239 Tensor b_tensor(DT_FLOAT, TensorShape({width})); in TestMergeDuplicateConsts() local
1240 test::FillIota<float>(&b_tensor, 1.0f); in TestMergeDuplicateConsts()
1241 Output b_op = Const(root.WithOpName("b_op"), Input::Initializer(b_tensor)); in TestMergeDuplicateConsts()
1286 Tensor b_tensor(DT_FLOAT, TensorShape({width})); in TestMergeDuplicatesNested() local
[all …]
|
| /external/tensorflow/tensorflow/core/debug/ |
| D | grpc_session_debug_test.cc | 60 Tensor b_tensor(DT_FLOAT, TensorShape({2, 1})); in CreateGraphDef() local
61 test::FillValues<float>(&b_tensor, {2.0, 1.0}); in CreateGraphDef()
62 Node* b = test::graph::Constant(&graph, b_tensor); in CreateGraphDef()
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | optional_ops.h | 145 const Tensor& b_tensor = b.get_values()[i]; in OptionalBinaryAdd() local
148 BinaryAddTensors<Device>(ctx, a_tensor, b_tensor, &out_tensor)); in OptionalBinaryAdd()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | message_wrappers_test.cc | 35 Tensor b_tensor(DT_INT32, TensorShape({1, 2})); in TensorB() local
36 test::FillValues<int32>(&b_tensor, {1, 2}); in TensorB()
37 return b_tensor; in TensorB()
|
| D | master_test.cc | 307 Tensor b_tensor(DT_FLOAT, TensorShape({2, 2})); in TEST_F() local
308 test::FillValues<float>(&b_tensor, {1, 0, 0, 1}); in TEST_F()
309 test::graph::Constant(&graph_1, b_tensor, "B"); in TEST_F()
|
| /external/tensorflow/tensorflow/python/training/ |
| D | monitored_session_test.py | 1350 b_tensor = constant_op.constant([0], name='b_tensor')
1351 add_tensor = a_tensor + b_tensor
1355 None, feed_dict={b_tensor: [10]})
1367 b_tensor = constant_op.constant([0], name='b_tensor')
1369 add_tensor = a_tensor + b_tensor + c_tensor
1373 None, feed_dict={b_tensor: [10]})
1389 b_tensor = constant_op.constant([0], name='b_tensor')
1390 add_tensor = a_tensor + b_tensor
1407 b_tensor = constant_op.constant([0], name='b_tensor')
1408 add_tensor = a_tensor + b_tensor
[all …]
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | grpc_session_test.cc | 52 Tensor b_tensor(DT_FLOAT, TensorShape({2, 1})); in CreateGraphDef() local
53 test::FillValues<float>(&b_tensor, {2, 1}); in CreateGraphDef()
54 Node* b = test::graph::Constant(&graph, b_tensor); in CreateGraphDef()
420 Tensor b_tensor(DT_FLOAT, TensorShape({kSize, 1})); in TEST() local
423 b_tensor.flat<float>()(i) = 3; in TEST()
426 Node* b = test::graph::Constant(&graph, b_tensor); in TEST()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | list_kernels.h | 863 const Tensor& b_tensor = b.tensors()[i]; in TensorListBinaryAdd() local
866 BinaryAddTensors<Device>(c, a_tensor, b_tensor, &out_tensor)); in TensorListBinaryAdd()
|