| /external/tensorflow/tensorflow/python/client/ |
| D | session_test.py | 651 indices_out, values_out, shape_out = sp_out
654 self.assertAllEqual(shape_out, shape)
661 indices_out, values_out, shape_out = s.run(sp)
664 self.assertAllEqual(shape_out, shape)
666 (indices_out, values_out, shape_out), = s.run([sp])
669 self.assertAllEqual(shape_out, shape)
676 indices_out, values_out, shape_out = s.run({'sp': sp})['sp']
679 self.assertAllEqual(shape_out, shape)
681 (indices_out, values_out, shape_out), = s.run({'sp': [sp]})['sp']
684 self.assertAllEqual(shape_out, shape)
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | sparse_cross_op.cc | 663 Tensor** values_out, Tensor** shape_out, in CreateOutputTensors() argument
681 TF_RETURN_IF_ERROR(context->allocate_output(2, TensorShape({2}), shape_out)); in CreateOutputTensors()
684 auto shape_vec = (*shape_out)->vec<int64>(); in CreateOutputTensors()
726 Tensor* shape_out; in Compute() local
732 &values_out, &shape_out, &output_start_indices)); in Compute()
788 Tensor* shape_out; in Compute() local
794 &values_out, &shape_out, &output_start_indices)); in Compute()
857 Tensor* shape_out; in Compute() local
863 &values_out, &shape_out, &output_start_indices)); in Compute()
|
| /external/tensorflow/tensorflow/c/experimental/saved_model/internal/ |
| D | saved_model_api_test.cc | 249 const TF_Shape* shape_out = TF_TensorSpecShape(tensor_spec_out); in TEST_P() local
254 EXPECT_EQ(0, TF_ShapeDims(shape_out)); in TEST_P()
|
| /external/tensorflow/tensorflow/core/grappler/ |
| D | grappler_item_builder.cc | 95 TensorShape* shape_out) { in ReplaceUnknownShapeDim() argument
108 return TensorShapeUtils::MakeShape(dims.data(), dims.size(), shape_out); in ReplaceUnknownShapeDim()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | sparse_ops.py | 651 indices_out, values_out, shape_out = gen_sparse_ops.sparse_cross_v2(
658 return sparse_tensor.SparseTensor(indices_out, values_out, shape_out)
777 indices_out, values_out, shape_out = gen_sparse_ops.sparse_cross(
789 return sparse_tensor.SparseTensor(indices_out, values_out, shape_out)
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | shape_inference_test.cc | 867 Shape shape_out = ShapeUtil::MakeShape(C64, {16, 5}); in TEST_F() local
869 fft::Pass(shape_in, type, {8}, shape_out); in TEST_F()
870 fft::Pass(shape_in, type, {16, 8}, shape_out); in TEST_F()
890 Shape shape_out = ShapeUtil::MakeShape(C64, {16, 5}); in TEST_F() local
891 fft::Pass(even_shape_in, type, {16, 8}, shape_out); in TEST_F()
892 fft::Pass(odd_shape_in, type, {16, 9}, shape_out); in TEST_F()
906 Shape shape_out = ShapeUtil::MakeShape(F32, {16, 8}); in TEST_F() local
908 fft::Pass(shape_in, type, {8}, shape_out); in TEST_F()
909 fft::Pass(shape_in, type, {16, 8}, shape_out); in TEST_F()
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | graph_properties_test.cc | 620 auto shape_out = properties.GetOutputProperties("while/Exit_1").at(0).shape(); in TEST_F() local
622 EXPECT_GE(-2, shape_out.dim(0).size()); in TEST_F()
623 EXPECT_NE(shape_in.dim(0).size(), shape_out.dim(0).size()); in TEST_F()
|