| /external/tensorflow/tensorflow/contrib/rnn/kernels/ |
| D | lstm_ops.cc | 265 const Tensor* b_tensor = nullptr; in Compute() local
266 OP_REQUIRES_OK(ctx, ctx->input("b", &b_tensor)); in Compute()
300 OP_REQUIRES(ctx, b_tensor->dim_size(0) == cell_size * 4, in Compute()
302 "b.dim_size(0) != cell_size * 4: ", b_tensor->dim_size(0), in Compute()
361 wcf_tensor->vec<T>(), wco_tensor->vec<T>(), b_tensor->vec<T>(), in Compute()
446 const Tensor* b_tensor = nullptr; in Compute() local
447 OP_REQUIRES_OK(ctx, ctx->input("b", &b_tensor)); in Compute()
505 OP_REQUIRES(ctx, b_tensor->dim_size(0) == cell_size * 4, in Compute()
507 "b.dim_size(0) != cell_size * 4: ", b_tensor->dim_size(0), in Compute()
646 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/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 | 306 Tensor b_tensor(DT_FLOAT, TensorShape({2, 2})); in TEST_F() local
307 test::FillValues<float>(&b_tensor, {1, 0, 0, 1}); in TEST_F()
308 test::graph::Constant(&graph_1, b_tensor, "B"); in TEST_F()
|
| /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/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()
385 Tensor b_tensor(DT_FLOAT, TensorShape({kSize, 1})); in TEST() local
388 b_tensor.flat<float>()(i) = 3; in TEST()
391 Node* b = test::graph::Constant(&graph, b_tensor); in TEST()
|
| /external/tensorflow/tensorflow/python/training/ |
| D | monitored_session_test.py | 1304 b_tensor = constant_op.constant([0], name='b_tensor')
1305 add_tensor = a_tensor + b_tensor
1309 None, feed_dict={b_tensor: [10]})
1321 b_tensor = constant_op.constant([0], name='b_tensor')
1323 add_tensor = a_tensor + b_tensor + c_tensor
1327 None, feed_dict={b_tensor: [10]})
1343 b_tensor = constant_op.constant([0], name='b_tensor')
1344 add_tensor = a_tensor + b_tensor
1361 b_tensor = constant_op.constant([0], name='b_tensor')
1362 add_tensor = a_tensor + b_tensor
[all …]
|
| /external/tensorflow/tensorflow/contrib/labeled_tensor/python/ops/ |
| D | ops.py | 861 b_tensor = array_ops.reshape(b.tensor, (-1, 1))
865 b_tensor = b.tensor
869 a_tensor, b_tensor, transpose_a=transpose_a, transpose_b=transpose_b)
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | list_kernels.h | 886 const Tensor& b_tensor = b.tensors[i]; in TensorListBinaryAdd() local
889 BinaryAddTensors<Device>(c, a_tensor, b_tensor, &out_tensor)); in TensorListBinaryAdd()
|