HomeSort by relevance Sort by last modified time
    Searched refs:const_tensor (Results 1 - 5 of 5) sorted by null

  /external/tensorflow/tensorflow/tools/graph_transforms/
insert_logging_test.cc 49 Tensor const_tensor(DT_FLOAT, TensorShape({10}));
50 test::FillIota<float>(&const_tensor, 1.0f);
52 Const(root.WithOpName("const_node1"), Input::Initializer(const_tensor));
54 Const(root.WithOpName("const_node2"), Input::Initializer(const_tensor));
56 Const(root.WithOpName("const_node3"), Input::Initializer(const_tensor));
96 Tensor const_tensor(DT_FLOAT, TensorShape({10}));
97 test::FillIota<float>(&const_tensor, 1.0f);
99 Const(root.WithOpName("const_node1"), Input::Initializer(const_tensor));
101 Const(root.WithOpName("const_node2"), Input::Initializer(const_tensor));
103 Const(root.WithOpName("const_node3"), Input::Initializer(const_tensor));
    [all...]
  /external/tensorflow/tensorflow/compiler/jit/kernels/
xla_launch_op.cc 336 const Tensor& const_tensor = kernel->outputs[i].constant_value; local
337 const size_t total_bytes = const_tensor.TotalBytes();
343 ctx->allocate_output(i, const_tensor.shape(), &output_tensor));
345 const void* src_ptr = DMAHelper::base(&const_tensor);
351 ctx->set_output(i, const_tensor);
  /external/tensorflow/tensorflow/core/kernels/
remote_fused_graph_execute_op_test.cc 188 Tensor const_tensor; variable
190 *proto, &const_tensor));
191 const float b_val = *const_tensor.scalar<float>().data();
  /external/tensorflow/tensorflow/core/kernels/hexagon/
graph_transferer.cc 488 Tensor const_tensor; local
489 TF_CHECK_OK(MakeTensorFromProto(*proto, &const_tensor));
491 const_node_info.set_dtype(const_tensor.dtype());
493 const_node_info.set_data(const_tensor.tensor_data().data(), data_size);
733 Tensor const_tensor; local
734 TF_CHECK_OK(MakeTensorFromProto(*tensor_proto, &const_tensor));
735 CHECK_EQ(2, const_tensor.shape().dims());
736 CHECK_EQ(PAD_HEIGHT, const_tensor.shape().dim_size(1));
737 if (const_tensor.shape().dim_size(0) == PAD_WIDTH) {
739 } else if (const_tensor.shape().dim_size(0) < PAD_WIDTH)
743 Tensor const_tensor; local
    [all...]
  /external/tensorflow/tensorflow/python/framework/
constant_op.py 216 const_tensor = g.create_op(
221 return const_tensor

Completed in 218 milliseconds