Home
last modified time | relevance | path

Searched refs:MakeShapeFromShapeTensor (Results 1 – 19 of 19) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/ops/
H A Dstateful_random_ops.cc30 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(2, &out)); in StatefulRandomShape()
78 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(2, &out)); in __anonb8aa84710102()
116 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(0, &out)); in __anonb8aa84710402()
135 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(2, &out)); in __anonb8aa84710502()
154 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(1, &out)); in __anonb8aa84710602()
H A Drandom_ops.cc145 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(0, &out)); in __anond8b80b8a0402()
169 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(0, &out)); in __anond8b80b8a0502()
188 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(0, &out)); in __anond8b80b8a0602()
H A Dnn_ops.cc409 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(1, &s)); in __anon84cbdd650402()
623 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(0, &s)); in __anon84cbdd650802()
642 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(1, &s)); in __anon84cbdd650902()
724 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(0, &s)); in __anon84cbdd650c02()
742 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(1, &s)); in __anon84cbdd650d02()
1566 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(0, &out)); in __anon84cbdd652302()
1936 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(1, &s)); in __anon84cbdd652a02()
1963 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(1, &s)); in __anon84cbdd652b02()
2002 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(1, &sh)); in __anon84cbdd652c02()
2079 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(0, &s)); in __anon84cbdd652d02()
[all …]
H A Dstateless_random_ops.cc34 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(0, &out)); in StatelessShape()
157 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(0, &out)); in __anonbacd0d810302()
H A Dmkl_nn_ops.cc67 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(0, &s)); in __anon8e35a5880102()
93 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(1, &s)); in __anon8e35a5880202()
178 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(0, &s)); in __anon8e35a5880302()
197 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(1, &s)); in __anon8e35a5880402()
1143 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(0, &s)); in __anon8e35a5881002()
1174 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(1, &s)); in __anon8e35a5881102()
H A Dsparse_ops.cc104 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(2, &a_shape)); in __anone91b66370402()
206 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(1, &out)); in __anone91b66370902()
632 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(2, &constant_input_shape)); in __anone91b66371802()
H A Dnccl_ops.cc132 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(0, &out)); in __anona6b06b490102()
H A Dstateless_random_ops_v2.cc39 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(0, &out)); in StatelessShapeV2()
H A Dcollective_ops.cc199 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(/*input_idx=*/3, &out)); in __anon66a0a2330402()
H A Darray_ops.cc191 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(1, &out)); in SetOutputShapeForReshape()
409 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(0, &out)); in __anon38bbb0e80402()
479 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(1, &out)); in __anon38bbb0e80702()
1121 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(0, &out)); in __anon38bbb0e81602()
1810 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(0, &out)); in __anon38bbb0e82602()
1877 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(1, &multiples)); in __anon38bbb0e82702()
3186 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(2, &output_shape)); in __anon38bbb0e84702()
H A Dsparse_csr_matrix_ops.cc70 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(2, &dense_shape)); in __anona88575120102()
460 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(0, &dense_shape)); in __anona88575120b02()
H A Dimage_ops.cc920 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(3, &out)); in __anon269972ec1702()
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Dshape_inference_test.cc234 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(0, &s)); in TEST_F()
235 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(1, &s)); in TEST_F()
260 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(0, &s)); in TEST_F()
261 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(1, &s)); in TEST_F()
1101 TEST_F(ShapeInferenceTest, MakeShapeFromShapeTensor) { in TEST_F() argument
1107 Status s = c.MakeShapeFromShapeTensor(0, &out); in TEST_F()
1166 c.MakeShapeFromShapeTensor(0, &out).error_message()); in TEST_F()
H A Dcommon_shape_fns.cc915 c->MakeShapeFromShapeTensor(0, &specified_input_grad_shape)); in Conv2DBackpropInputShape()
970 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(1, &sh)); in Conv2DBackpropFilterWithBiasShape()
1180 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(0, &s)); in AvgPoolGradShape()
1883 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(0, &s)); in AvgPool3DGradShape()
2175 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(0, &out)); in RandomShape()
2260 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(1, &begin_value)); in SliceShape()
2294 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(2, &sizes_value)); in SliceShape()
H A Dshape_inference.h550 Status MakeShapeFromShapeTensor(int input_idx, ShapeHandle* out);
568 StatusOr<ShapeHandle> MakeShapeFromShapeTensor(const TensorShape& shape);
H A Dshape_inference.cc743 Status InferenceContext::MakeShapeFromShapeTensor(int input_idx, in MakeShapeFromShapeTensor() function in tensorflow::shape_inference::InferenceContext
904 StatusOr<ShapeHandle> InferenceContext::MakeShapeFromShapeTensor( in MakeShapeFromShapeTensor() function in tensorflow::shape_inference::InferenceContext
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/ops/
H A Dxla_ops.cc394 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(2, &size_indices_value)); in __anon6b9e6b800202()
829 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(2, &output)); in __anon6b9e6b800b02()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Dlight_outside_compilation.cc139 ic.MakeShapeFromShapeTensor(shape)); in CompileToCustomCallCallingTfKernel()
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dshape_refiner_test.cc835 TF_RETURN_IF_ERROR(c->MakeShapeFromShapeTensor(0 /* input_idx */, &out)); in TensorAsShapeShapeFn()