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.cc282 TF_RETURN_IF_ERROR(c->Subshape(handle_shape_and_type[0].shape, 0, in __anon620315450402()
286 TF_RETURN_IF_ERROR(c->Subshape(handle_shape_and_type[0].shape, in __anon620315450402()
291 c->Subshape(indices_shape, batch_dims, &indices_subshape)); in __anon620315450402()
332 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 __anon6adce4610702()
281 TF_RETURN_IF_ERROR(c->Subshape(b_shape, 0, -2, &b_batch_dims)); in __anon6adce4610702()
433 TF_RETURN_IF_ERROR(c->Subshape(a_shape, 0, -2, &a_batch_dims)); in __anon6adce4610a02()
434 TF_RETURN_IF_ERROR(c->Subshape(b_shape, 0, -2, &b_batch_dims)); in __anon6adce4610a02()
598 TF_RETURN_IF_ERROR(c->Subshape(matrix_shape, 0, -2, &matrix_batch_shape)); in __anon6adce4611002()
599 TF_RETURN_IF_ERROR(c->Subshape(perm_shape, 0, -1, &perm_shape)); in __anon6adce4611002()
Dset_ops.cc66 c->Subshape(input0_shape, 0, rank - 1, &group0_shape)); in __anon1e42d5040102()
69 c->Subshape(input1_shape, 0, rank - 1, &group1_shape)); in __anon1e42d5040102()
Dencode_proto_ops.cc41 TF_RETURN_IF_ERROR(c->Subshape(input, 0, -1, &inner)); in __anon8a6985790102()
Dlist_ops.cc290 TF_RETURN_IF_ERROR(c->Subshape(element_shape, 1, &result)); in TensorListConcatShapeInference()
352 c->Subshape(tensor_shape, 1, &element_shape_from_tensor_shape)); in __anon6c1e81940902()
385 c->Subshape(tensor_shape, 1, &tensor_shape_except_first_dim)); in __anon6c1e81940a02()
Dragged_array_ops.cc166 TF_RETURN_IF_ERROR(c->Subshape(params_dense_values, 1, &value)); in RaggedGatherShapeFn()
Dcollective_ops.cc57 TF_RETURN_IF_ERROR(c->Subshape(c->input(0), 1, &in_subshape)); in __anone3133c620102()
Darray_ops.cc898 TF_RETURN_IF_ERROR(c->Subshape(diag, 0, -1, &diag_batch_shape)); in __anon091a71b71102()
1186 TF_RETURN_IF_ERROR(c->Subshape(c->input(0), 1, &params_subshape)); in __anon091a71b71702()
1256 c->Subshape(params_shape, 0, axis, &params_outer_subshape)); in __anon091a71b71802()
1260 c->Subshape(indices_shape, batch_dims, &indices_inner_subshape)); in __anon091a71b71802()
1273 c->Subshape(params_shape, axis + 1, &params_inner_subshape)); in __anon091a71b71802()
2146 TF_RETURN_IF_ERROR(c->Subshape(input, dim, &end)); in __anon091a71b72f02()
2150 TF_RETURN_IF_ERROR(c->Subshape(input, 0, dim, &output)); in __anon091a71b72f02()
2333 c->Subshape(input_shape, 1 + num_block_dims, &remaining_input_shape)); in SpaceToBatchShapeHelper()
2408 c->Subshape(input_shape, 1 + num_block_dims, &remaining_input_shape)); in BatchToSpaceShapeHelper()
2834 TF_RETURN_IF_ERROR(c->Subshape(indices, 0, depth_index, &front)); in __anon091a71b73b02()
[all …]
Dstate_ops.cc123 TF_RETURN_IF_ERROR(c->Subshape(var_shape, 1, &var_subshape)); in ScatterUpdateShape()
Dmath_ops.cc1162 TF_RETURN_IF_ERROR(c->Subshape(data_shape, 1, &subshape)); in SegmentReductionShapeFn()
1186 TF_RETURN_IF_ERROR(c->Subshape(data_shape, 1, &subshape)); in SparseSegmentReductionShapeFn()
1210 TF_RETURN_IF_ERROR(c->Subshape(data_shape, 1, &subshape)); in SparseSegmentReductionGradShapeFn()
1251 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 __anon89875fc90202()
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 __anon89875fc91b02()
Dnn_ops.cc1354 TF_RETURN_IF_ERROR(c->Subshape(input, 0, -1, &s)); in TopKShapeFn()
1505 TF_RETURN_IF_ERROR(c->Subshape(input, 0, -1, &s)); in __anon1bbab8942102()
/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()
1570 TF_RETURN_IF_ERROR(c->Subshape( in MatrixSetDiagV2Shape()
2030 TF_RETURN_IF_ERROR(c->Subshape(input, 0, concat_dim, &output_before)); in ConcatShapeHelper()
[all …]
Dshape_inference_test.cc844 TEST_F(ShapeInferenceTest, Subshape) { in TEST_F() argument
851 TF_EXPECT_OK(c.Subshape(unknown, 0, &out)); in TEST_F()
854 TF_EXPECT_OK(c.Subshape(unknown, 1, &out)); in TEST_F()
857 TF_EXPECT_OK(c.Subshape(unknown, 200, &out)); in TEST_F()
864 TF_EXPECT_OK(c.Subshape(in0, 0, &out)); in TEST_F()
875 TF_ASSERT_OK(c.Subshape(in0, start, end, &out_arr[0])); in TEST_F()
876 TF_ASSERT_OK(c.Subshape(in0, neg_start, end, &out_arr[1])); in TEST_F()
877 TF_ASSERT_OK(c.Subshape(in0, start, neg_end, &out_arr[2])); in TEST_F()
878 TF_ASSERT_OK(c.Subshape(in0, neg_start, neg_end, &out_arr[3])); in TEST_F()
898 c.Subshape(in0, 6, -3, &out), in TEST_F()
[all …]
Dshape_inference.h501 Status Subshape(ShapeHandle s, int64_t start,
507 Status Subshape(ShapeHandle s, int64_t start, int64_t end,
514 Status Subshape(ShapeHandle s, int64_t start, int64_t end, int64_t stride,
Dshape_inference.cc575 Status InferenceContext::Subshape(ShapeHandle s, int64_t start, in Subshape() function in tensorflow::shape_inference::InferenceContext
577 return Subshape(s, start, std::numeric_limits<int64_t>::max() /* end */, out); in Subshape()
580 Status InferenceContext::Subshape(ShapeHandle s, int64_t start, int64_t end, in Subshape() function in tensorflow::shape_inference::InferenceContext
582 return Subshape(s, start, end, 1 /* stride */, out); in Subshape()
585 Status InferenceContext::Subshape(ShapeHandle s, int64_t start, int64_t 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 __anoncc7f7f1e0102()
/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.cc664 TF_RETURN_IF_ERROR(ctx->Subshape(input, begin, end, stride, result)); in PartialStridedSliceShape()
/external/tensorflow/tensorflow/core/grappler/costs/
Dgraph_properties.cc1836 TF_RETURN_IF_ERROR(ic->Subshape(input, start, &result)); in MaybeUpdateNodeContextOutput()
1839 TF_RETURN_IF_ERROR(ic->Subshape(input, start, end, &result)); in MaybeUpdateNodeContextOutput()
1894 TF_RETURN_IF_ERROR(ic->Subshape(input, begin, end, stride, &result)); in MaybeUpdateNodeContextOutput()