Searched refs:const_tensor (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | insert_logging_test.cc | 49 Tensor const_tensor(DT_FLOAT, TensorShape({10})); in TestInsertLogging() local
50 test::FillIota<float>(&const_tensor, 1.0f); in TestInsertLogging()
52 Const(root.WithOpName("const_node1"), Input::Initializer(const_tensor)); in TestInsertLogging()
54 Const(root.WithOpName("const_node2"), Input::Initializer(const_tensor)); in TestInsertLogging()
56 Const(root.WithOpName("const_node3"), Input::Initializer(const_tensor)); in TestInsertLogging()
96 Tensor const_tensor(DT_FLOAT, TensorShape({10})); in TestInsertLoggingByOpType() local
97 test::FillIota<float>(&const_tensor, 1.0f); in TestInsertLoggingByOpType()
99 Const(root.WithOpName("const_node1"), Input::Initializer(const_tensor)); in TestInsertLoggingByOpType()
101 Const(root.WithOpName("const_node2"), Input::Initializer(const_tensor)); in TestInsertLoggingByOpType()
103 Const(root.WithOpName("const_node3"), Input::Initializer(const_tensor)); in TestInsertLoggingByOpType()
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/hexagon/ |
| D | graph_transferer.cc | 495 Tensor const_tensor; in RegisterConstantNode() local
496 TF_CHECK_OK(MakeTensorFromProto(*proto, &const_tensor)); in RegisterConstantNode()
498 const_node_info.set_dtype(const_tensor.dtype()); in RegisterConstantNode()
500 const_node_info.set_data(const_tensor.tensor_data().data(), data_size); in RegisterConstantNode()
740 Tensor const_tensor; in RegisterPadNode() local
741 TF_CHECK_OK(MakeTensorFromProto(*tensor_proto, &const_tensor)); in RegisterPadNode()
742 CHECK_EQ(2, const_tensor.shape().dims()); in RegisterPadNode()
743 CHECK_EQ(PAD_HEIGHT, const_tensor.shape().dim_size(1)); in RegisterPadNode()
744 if (const_tensor.shape().dim_size(0) == PAD_WIDTH) { in RegisterPadNode()
746 } else if (const_tensor.shape().dim_size(0) < PAD_WIDTH) { in RegisterPadNode()
[all …]
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_launch_util.cc | 304 const Tensor& const_tensor = kernel->outputs[i].constant_value; in PopulateOutputs() local
306 const size_t total_bytes = const_tensor.TotalBytes(); in PopulateOutputs()
316 ctx->allocate_output(i, const_tensor.shape(), &output_tensor)); in PopulateOutputs()
323 &const_tensor, device, output_tensor, in PopulateOutputs()
338 ctx->set_output(i, const_tensor); in PopulateOutputs()
342 xla_tensor->set_host_tensor(const_tensor); in PopulateOutputs()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | remote_fused_graph_execute_op_test.cc | 188 Tensor const_tensor; in ExecuteGraph() local
190 *proto, &const_tensor)); in ExecuteGraph()
191 const float b_val = *const_tensor.scalar<float>().data(); in ExecuteGraph()
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | constant_op.py | 285 const_tensor = g.create_op(
290 return const_tensor
|
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | wrappers_test.py | 577 const_tensor = constant_op.constant(1)
580 layer.forward_layer.add_loss(const_tensor, inputs=None)
582 layer.backward_layer.add_loss(const_tensor, inputs=None)
|
| /external/tensorflow/tensorflow/python/data/kernel_tests/ |
| D | map_test.py | 883 const_tensor = ops.get_default_graph().create_op(
890 return const_tensor
|