| /external/tensorflow/tensorflow/core/kernels/ |
| list_kernels.h | 204 const Tensor& b_tensor = b.tensors[i]; local
206 out->tensors.push_back(b_tensor);
209 if (b_tensor.dtype() == DT_INVALID) {
213 if (a_tensor.shape() != b_tensor.shape()) {
219 b_tensor.shape().DebugString(), " in position ", i);
229 a_tensor.flat<dtype>() + b_tensor.flat<dtype>(); \
|
| /external/tensorflow/tensorflow/core/debug/ |
| grpc_session_debug_test.cc | 60 Tensor b_tensor(DT_FLOAT, TensorShape({2, 1}));
61 test::FillValues<float>(&b_tensor, {2.0, 1.0});
62 Node* b = test::graph::Constant(&graph, b_tensor);
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| message_wrappers_test.cc | 35 Tensor b_tensor(DT_INT32, TensorShape({1, 2}));
36 test::FillValues<int32>(&b_tensor, {1, 2});
37 return b_tensor;
|
| master_test.cc | 306 Tensor b_tensor(DT_FLOAT, TensorShape({2, 2}));
307 test::FillValues<float>(&b_tensor, {1, 0, 0, 1});
308 test::graph::Constant(&graph_1, b_tensor, "B");
|
| /external/tensorflow/tensorflow/contrib/rnn/kernels/ |
| lstm_ops.cc | 262 const Tensor* b_tensor = nullptr; variable
263 OP_REQUIRES_OK(ctx, ctx->input("b", &b_tensor));
297 OP_REQUIRES(ctx, b_tensor->dim_size(0) == cell_size * 4,
299 "b.dim_size(0) != cell_size * 4: ", b_tensor->dim_size(0),
358 wcf_tensor->vec<T>(), wco_tensor->vec<T>(), b_tensor->vec<T>(),
441 const Tensor* b_tensor = nullptr; variable
442 OP_REQUIRES_OK(ctx, ctx->input("b", &b_tensor));
500 OP_REQUIRES(ctx, b_tensor->dim_size(0) == cell_size * 4,
502 "b.dim_size(0) != cell_size * 4: ", b_tensor->dim_size(0),
641 wco_tensor->vec<T>(), b_tensor->vec<T>(), i_tensor->matrix<T>()
907 const Tensor* b_tensor = nullptr; variable
1081 const Tensor* b_tensor = nullptr; variable
[all...] |
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| quantize_nodes_test.cc | 217 Tensor b_tensor(DT_FLOAT, TensorShape({k, n}));
218 test::FillValues<float>(&b_tensor, b_values);
219 Output b_op = Const(root.WithOpName("b_op"), Input::Initializer(b_tensor));
380 Tensor b_tensor(DT_FLOAT, TensorShape({2, 2, 3}));
381 test::FillValues<float>(&b_tensor,
383 Output b_op = Const(root.WithOpName("b_op"), Input::Initializer(b_tensor));
[all...] |
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| grpc_session_test.cc | 52 Tensor b_tensor(DT_FLOAT, TensorShape({2, 1}));
53 test::FillValues<float>(&b_tensor, {2, 1});
54 Node* b = test::graph::Constant(&graph, b_tensor);
309 Tensor b_tensor(DT_FLOAT, TensorShape({kSize, 1}));
312 b_tensor.flat<float>()(i) = 3;
315 Node* b = test::graph::Constant(&graph, b_tensor);
[all...] |
| /external/tensorflow/tensorflow/python/training/ |
| monitored_session_test.py | [all...] |
| /external/tensorflow/tensorflow/contrib/labeled_tensor/python/ops/ |
| ops.py | 859 b_tensor = array_ops.reshape(b.tensor, (-1, 1))
863 b_tensor = b.tensor
867 a_tensor, b_tensor, transpose_a=transpose_a, transpose_b=transpose_b)
[all...] |
