Home
last modified time | relevance | path

Searched refs:Subshape (Results 1 – 21 of 21) sorted by relevance

/external/tensorflow/tensorflow/core/ops/
Dlinalg_ops.cc38 TF_RETURN_IF_ERROR(c->Subshape(s, 0, -2, &batch_shape)); in MakeBatchSquareMatrix()
76 TF_RETURN_IF_ERROR(c->Subshape(lhs, 0, -2, &lhs_batch_shape)); in BandedTriangularSolveShapeFn()
77 TF_RETURN_IF_ERROR(c->Subshape(rhs, 0, -2, &rhs_batch_shape)); in BandedTriangularSolveShapeFn()
108 TF_RETURN_IF_ERROR(c->Subshape(lhs, 0, -2, &lhs_batch_shape)); in MatrixSolveShapeFn()
109 TF_RETURN_IF_ERROR(c->Subshape(rhs, 0, -2, &rhs_batch_shape)); in MatrixSolveShapeFn()
142 TF_RETURN_IF_ERROR(c->Subshape(lhs, 0, -2, &lhs_batch_shape)); in MatrixTriangularSolveShapeFn()
143 TF_RETURN_IF_ERROR(c->Subshape(rhs, 0, -2, &rhs_batch_shape)); in MatrixTriangularSolveShapeFn()
167 TF_RETURN_IF_ERROR(c->Subshape(input, 0, -2, &batch_shape)); in SelfAdjointEigV2ShapeFn()
194 TF_RETURN_IF_ERROR(c->Subshape(input, 0, -2, &batch_shape)); in LuShapeFn()
220 TF_RETURN_IF_ERROR(c->Subshape(input, 0, -2, &batch_shape)); in QrShapeFn()
[all …]
Dresource_variable_ops.cc266 TF_RETURN_IF_ERROR(c->Subshape(handle_shape_and_type[0].shape, 0, in __anon50d007a70402()
270 TF_RETURN_IF_ERROR(c->Subshape(handle_shape_and_type[0].shape, in __anon50d007a70402()
275 c->Subshape(indices_shape, batch_dims, &indices_subshape)); in __anon50d007a70402()
316 TF_RETURN_IF_ERROR(c->Subshape(var_shape, 1, &var_subshape)); in ResourceScatterUpdateShape()
Dsparse_csr_matrix_ops.cc280 TF_RETURN_IF_ERROR(c->Subshape(a_shape, 0, -2, &a_batch_dims)); in __anon59a9d6c30702()
281 TF_RETURN_IF_ERROR(c->Subshape(b_shape, 0, -2, &b_batch_dims)); in __anon59a9d6c30702()
433 TF_RETURN_IF_ERROR(c->Subshape(a_shape, 0, -2, &a_batch_dims)); in __anon59a9d6c30a02()
434 TF_RETURN_IF_ERROR(c->Subshape(b_shape, 0, -2, &b_batch_dims)); in __anon59a9d6c30a02()
598 TF_RETURN_IF_ERROR(c->Subshape(matrix_shape, 0, -2, &matrix_batch_shape)); in __anon59a9d6c31002()
599 TF_RETURN_IF_ERROR(c->Subshape(perm_shape, 0, -1, &perm_shape)); in __anon59a9d6c31002()
Dencode_proto_ops.cc41 TF_RETURN_IF_ERROR(c->Subshape(input, 0, -1, &inner)); in __anona4faf29b0102()
Dset_ops.cc66 c->Subshape(input0_shape, 0, rank - 1, &group0_shape)); in __anonda6063e60102()
69 c->Subshape(input1_shape, 0, rank - 1, &group1_shape)); in __anonda6063e60102()
Dcollective_ops.cc55 TF_RETURN_IF_ERROR(c->Subshape(c->input(0), 1, &in_subshape)); in __anon52c2f9040102()
Dragged_array_ops.cc152 TF_RETURN_IF_ERROR(c->Subshape(params_dense_values, 1, &value)); in RaggedGatherShapeFn()
Dlist_ops.cc276 TF_RETURN_IF_ERROR(c->Subshape(element_shape, 1, &result)); in TensorListConcatShapeInference()
336 c->Subshape(tensor_shape, 1, &element_shape_from_tensor_shape)); in __anonabedecb60902()
365 c->Subshape(tensor_shape, 1, &tensor_shape_except_first_dim)); in __anonabedecb60a02()
Darray_ops.cc880 TF_RETURN_IF_ERROR(c->Subshape(diag, 0, -1, &diag_batch_shape)); in __anon42d741191102()
1167 TF_RETURN_IF_ERROR(c->Subshape(c->input(0), 1, &params_subshape)); in __anon42d741191702()
1234 c->Subshape(params_shape, 0, axis, &params_outer_subshape)); in __anon42d741191802()
1238 c->Subshape(indices_shape, batch_dims, &indices_inner_subshape)); in __anon42d741191802()
1251 c->Subshape(params_shape, axis + 1, &params_inner_subshape)); in __anon42d741191802()
2113 TF_RETURN_IF_ERROR(c->Subshape(input, dim, &end)); in __anon42d741192f02()
2117 TF_RETURN_IF_ERROR(c->Subshape(input, 0, dim, &output)); in __anon42d741192f02()
2300 c->Subshape(input_shape, 1 + num_block_dims, &remaining_input_shape)); in SpaceToBatchShapeHelper()
2375 c->Subshape(input_shape, 1 + num_block_dims, &remaining_input_shape)); in BatchToSpaceShapeHelper()
2801 TF_RETURN_IF_ERROR(c->Subshape(indices, 0, depth_index, &front)); in __anon42d741193b02()
[all …]
Dstate_ops.cc123 TF_RETURN_IF_ERROR(c->Subshape(var_shape, 1, &var_subshape)); in ScatterUpdateShape()
Dmath_ops.cc1129 TF_RETURN_IF_ERROR(c->Subshape(data_shape, 1, &subshape)); in SegmentReductionShapeFn()
1153 TF_RETURN_IF_ERROR(c->Subshape(data_shape, 1, &subshape)); in SparseSegmentReductionShapeFn()
1177 TF_RETURN_IF_ERROR(c->Subshape(data_shape, 1, &subshape)); in SparseSegmentReductionGradShapeFn()
1218 TF_RETURN_IF_ERROR(c->Subshape(data_shape, 1, &subshape)); in SparseSegmentReductionWithNumSegmentsShapeFn()
Ddata_flow_ops.cc77 TF_RETURN_IF_ERROR(c->Subshape(data_shape, rank, &data_suffix_shape)); in __anonea01102b0202()
118 TF_RETURN_IF_ERROR(c->Subshape(data_shape, indices_rank, &rest)); in DynamicStitchShapeFunction()
823 TF_RETURN_IF_ERROR(c->Subshape(value_shape, 1, &fed_shape)); in __anonea01102b1b02()
Dnn_ops.cc1320 TF_RETURN_IF_ERROR(c->Subshape(input, 0, -1, &s)); in TopKShapeFn()
1376 TF_RETURN_IF_ERROR(c->Subshape(input, 0, -1, &s)); in __anonad10f3362002()
/external/tensorflow/tensorflow/core/framework/
Dcommon_shape_fns.cc225 TF_RETURN_IF_ERROR(c->Subshape(input_shape, axis_before_ellipsis, in EinsumShape()
332 TF_RETURN_IF_ERROR(c->Subshape(a_shape, 0, -2, &a_batch_shape)); in BatchMatMulV2Shape()
333 TF_RETURN_IF_ERROR(c->Subshape(b_shape, 0, -2, &b_batch_shape)); in BatchMatMulV2Shape()
364 TF_RETURN_IF_ERROR(c->Subshape(a_shape, 0, -2, &a_batch_dims)); in BatchMatMulShape()
365 TF_RETURN_IF_ERROR(c->Subshape(b_shape, 0, -2, &b_batch_dims)); in BatchMatMulShape()
411 TF_RETURN_IF_ERROR(c->Subshape(input_shape, 0, 1, &first)); in BiasAddShape()
414 TF_RETURN_IF_ERROR(c->Subshape(input_shape, 2, &last)); in BiasAddShape()
426 TF_RETURN_IF_ERROR(c->Subshape(input_shape, 0, -1, &all_but_bias)); in BiasAddShape()
1479 TF_RETURN_IF_ERROR(c->Subshape( in MatrixSetDiagV2Shape()
1939 TF_RETURN_IF_ERROR(c->Subshape(input, 0, concat_dim, &output_before)); in ConcatShapeHelper()
[all …]
Dshape_inference_test.cc824 TEST_F(ShapeInferenceTest, Subshape) { in TEST_F() argument
831 EXPECT_TRUE(c.Subshape(unknown, 0, &out).ok()); in TEST_F()
834 EXPECT_TRUE(c.Subshape(unknown, 1, &out).ok()); in TEST_F()
837 EXPECT_TRUE(c.Subshape(unknown, 200, &out).ok()); in TEST_F()
844 EXPECT_TRUE(c.Subshape(in0, 0, &out).ok()); in TEST_F()
855 ASSERT_TRUE(c.Subshape(in0, start, end, &out_arr[0]).ok()); in TEST_F()
856 ASSERT_TRUE(c.Subshape(in0, neg_start, end, &out_arr[1]).ok()); in TEST_F()
857 ASSERT_TRUE(c.Subshape(in0, start, neg_end, &out_arr[2]).ok()); in TEST_F()
858 ASSERT_TRUE(c.Subshape(in0, neg_start, neg_end, &out_arr[3]).ok()); in TEST_F()
878 c.Subshape(in0, 6, -3, &out).ToString(), in TEST_F()
[all …]
Dshape_inference.h416 Status Subshape(ShapeHandle s, int64 start,
422 Status Subshape(ShapeHandle s, int64 start, int64 end,
429 Status Subshape(ShapeHandle s, int64 start, int64 end, int64 stride,
Dshape_inference.cc568 Status InferenceContext::Subshape(ShapeHandle s, int64 start, in Subshape() function in tensorflow::shape_inference::InferenceContext
570 return Subshape(s, start, std::numeric_limits<int64>::max() /* end */, out); in Subshape()
573 Status InferenceContext::Subshape(ShapeHandle s, int64 start, int64 end, in Subshape() function in tensorflow::shape_inference::InferenceContext
575 return Subshape(s, start, end, 1 /* stride */, out); in Subshape()
578 Status InferenceContext::Subshape(ShapeHandle s, int64 start, int64 end, in Subshape() function in tensorflow::shape_inference::InferenceContext
/external/tensorflow/tensorflow/core/tpu/ops/
Dtopk_ops.cc33 TF_RETURN_IF_ERROR(c->Subshape(input, 0, -1, &s)); in __anonee73ab400102()
/external/tensorflow/tensorflow/c/
Dops.cc263 Status s = cc_ctx->Subshape(*reinterpret_cast<ShapeHandle*>(shape_handle), in TF_ShapeInferenceContextSubshape()
/external/tensorflow/tensorflow/core/common_runtime/
Dshape_refiner.cc643 TF_RETURN_IF_ERROR(ctx->Subshape(input, begin, end, stride, result)); in PartialStridedSliceShape()
/external/tensorflow/tensorflow/core/grappler/costs/
Dgraph_properties.cc1797 TF_RETURN_IF_ERROR(ic->Subshape(input, start, &result)); in MaybeUpdateNodeContextOutput()
1800 TF_RETURN_IF_ERROR(ic->Subshape(input, start, end, &result)); in MaybeUpdateNodeContextOutput()
1855 TF_RETURN_IF_ERROR(ic->Subshape(input, begin, end, stride, &result)); in MaybeUpdateNodeContextOutput()