Home
last modified time | relevance | path

Searched refs:shape_out (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/python/client/
Dsession_test.py626 indices_out, values_out, shape_out = sp_out
629 self.assertAllEqual(shape_out, shape)
636 indices_out, values_out, shape_out = s.run(sp)
639 self.assertAllEqual(shape_out, shape)
641 (indices_out, values_out, shape_out), = s.run([sp])
644 self.assertAllEqual(shape_out, shape)
651 indices_out, values_out, shape_out = s.run({'sp': sp})['sp']
654 self.assertAllEqual(shape_out, shape)
656 (indices_out, values_out, shape_out), = s.run({'sp': [sp]})['sp']
659 self.assertAllEqual(shape_out, shape)
[all …]
/external/tensorflow/tensorflow/core/kernels/
Dsparse_cross_op.cc322 Tensor* shape_out; in Compute() local
326 &shape_out, &output_start_indices); in Compute()
507 Tensor** values_out, Tensor** shape_out, in CreateOutputTensors() argument
528 context->allocate_output(2, TensorShape({2}), shape_out)); in CreateOutputTensors()
531 auto shape_vec = (*shape_out)->vec<int64>(); in CreateOutputTensors()
/external/tensorflow/tensorflow/core/grappler/
Dgrappler_item_builder.cc94 TensorShape* shape_out) { in ReplaceUnknownShapeDim() argument
107 return TensorShapeUtils::MakeShape(dims.data(), dims.size(), shape_out); in ReplaceUnknownShapeDim()
/external/tensorflow/tensorflow/compiler/xla/service/
Dshape_inference_test.cc681 Shape shape_out = ShapeUtil::MakeShape(C64, {16, 5}); in TEST_F() local
683 fft::Pass(shape_in, type, {8}, shape_out); in TEST_F()
684 fft::Pass(shape_in, type, {16, 8}, shape_out); in TEST_F()
704 Shape shape_out = ShapeUtil::MakeShape(C64, {16, 5}); in TEST_F() local
705 fft::Pass(even_shape_in, type, {16, 8}, shape_out); in TEST_F()
706 fft::Pass(odd_shape_in, type, {16, 9}, shape_out); in TEST_F()
720 Shape shape_out = ShapeUtil::MakeShape(F32, {16, 8}); in TEST_F() local
722 fft::Pass(shape_in, type, {8}, shape_out); in TEST_F()
723 fft::Pass(shape_in, type, {16, 8}, shape_out); in TEST_F()
/external/tensorflow/tensorflow/python/ops/
Dsparse_ops.py643 indices_out, values_out, shape_out = gen_sparse_ops.sparse_cross(
655 return sparse_tensor.SparseTensor(indices_out, values_out, shape_out)
/external/tensorflow/tensorflow/core/grappler/costs/
Dgraph_properties_test.cc598 auto shape_out = properties.GetOutputProperties("while/Exit_1").at(0).shape(); in TEST_F() local
600 EXPECT_GE(-2, shape_out.dim(0).size()); in TEST_F()
601 EXPECT_NE(shape_in.dim(0).size(), shape_out.dim(0).size()); in TEST_F()