Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/ops/
Dlinalg_ops.cc38 TF_RETURN_IF_ERROR(c->Subshape(s, 0, -2, &batch_shape)); in MakeBatchSquareMatrix()
65 TF_RETURN_IF_ERROR(c->Subshape(lhs, 0, -2, &lhs_batch_shape)); in MatrixSolveShapeFn()
66 TF_RETURN_IF_ERROR(c->Subshape(rhs, 0, -2, &rhs_batch_shape)); in MatrixSolveShapeFn()
96 TF_RETURN_IF_ERROR(c->Subshape(input, 0, -2, &batch_shape)); in SelfAdjointEigV2ShapeFn()
123 TF_RETURN_IF_ERROR(c->Subshape(input, 0, -2, &batch_shape)); in LuShapeFn()
149 TF_RETURN_IF_ERROR(c->Subshape(input, 0, -2, &batch_shape)); in QrShapeFn()
180 TF_RETURN_IF_ERROR(c->Subshape(input, 0, -2, &batch_shape)); in SvdShapeFn()
223 TF_RETURN_IF_ERROR(c->Subshape(lhs, 0, -2, &lhs_batch_shape)); in TridiagonalSolveShapeFn()
224 TF_RETURN_IF_ERROR(c->Subshape(rhs, 0, -2, &rhs_batch_shape)); in TridiagonalSolveShapeFn()
256 TF_RETURN_IF_ERROR(c->Subshape(input, 0, -2, &out)); in __anon4fdd0bd30202()
[all …]
Dresource_variable_ops.cc283 TF_RETURN_IF_ERROR(c->Subshape(handle_shape_and_type[0].shape, 0, in __anon3f9cfa090402()
287 TF_RETURN_IF_ERROR(c->Subshape(handle_shape_and_type[0].shape, in __anon3f9cfa090402()
292 c->Subshape(indices_shape, batch_dims, &indices_subshape)); in __anon3f9cfa090402()
324 TF_RETURN_IF_ERROR(c->Subshape(var_shape, 1, &var_subshape)); in ResourceScatterUpdateShape()
Dset_ops.cc66 c->Subshape(input0_shape, 0, rank - 1, &group0_shape)); in __anon967df2c80102()
69 c->Subshape(input1_shape, 0, rank - 1, &group1_shape)); in __anon967df2c80102()
Dencode_proto_ops.cc41 TF_RETURN_IF_ERROR(c->Subshape(input, 0, -1, &inner)); in __anonbf8c5fbd0102()
Darray_ops.cc851 TF_RETURN_IF_ERROR(c->Subshape(diag, 0, -1, &diag_prefix)); in __anon7c94107b1102()
1100 TF_RETURN_IF_ERROR(c->Subshape(c->input(0), 1, &params_subshape)); in __anon7c94107b1702()
1153 c->Subshape(params_shape, 0, axis, &params_outer_subshape)); in __anon7c94107b1802()
1166 c->Subshape(params_shape, axis + 1, &params_inner_subshape)); in __anon7c94107b1802()
1202 TF_RETURN_IF_ERROR(c->Subshape(indices, 0, -1, &indices_slice)); in __anon7c94107b1902()
1203 TF_RETURN_IF_ERROR(c->Subshape(params, c->Value(r_dim), &params_slice)); in __anon7c94107b1902()
1958 TF_RETURN_IF_ERROR(c->Subshape(input, dim, &end)); in __anon7c94107b2f02()
1962 TF_RETURN_IF_ERROR(c->Subshape(input, 0, dim, &output)); in __anon7c94107b2f02()
2143 c->Subshape(input_shape, 1 + num_block_dims, &remaining_input_shape)); in SpaceToBatchShapeHelper()
2218 c->Subshape(input_shape, 1 + num_block_dims, &remaining_input_shape)); in BatchToSpaceShapeHelper()
[all …]
Dragged_array_ops.cc78 TF_RETURN_IF_ERROR(c->Subshape(params_dense_values, 1, &value)); in RaggedGatherShapeFn()
Dmath_ops.cc145 TF_RETURN_IF_ERROR(c->Subshape(a_shape, 0, -2, &a_batch_dims)); in __anonb22bfa860202()
146 TF_RETURN_IF_ERROR(c->Subshape(b_shape, 0, -2, &b_batch_dims)); in __anonb22bfa860202()
992 TF_RETURN_IF_ERROR(c->Subshape(data_shape, 1, &subshape)); in SegmentReductionShapeFn()
1016 TF_RETURN_IF_ERROR(c->Subshape(data_shape, 1, &subshape)); in SparseSegmentReductionShapeFn()
1040 TF_RETURN_IF_ERROR(c->Subshape(data_shape, 1, &subshape)); in SparseSegmentReductionGradShapeFn()
1081 TF_RETURN_IF_ERROR(c->Subshape(data_shape, 1, &subshape)); in SparseSegmentReductionWithNumSegmentsShapeFn()
1123 c->Subshape(s_data, c->Rank(s_segment_ids), &s_data_suffix)); in UnsortedSegmentReductionShapeFn()
Dlist_ops.cc265 TF_RETURN_IF_ERROR(c->Subshape(element_shape, 1, &result)); in TensorListConcatShapeInference()
325 c->Subshape(tensor_shape, 1, &element_shape_from_tensor_shape)); in __anonebbd57d80902()
354 c->Subshape(tensor_shape, 1, &tensor_shape_except_first_dim)); in __anonebbd57d80a02()
Dstate_ops.cc123 TF_RETURN_IF_ERROR(c->Subshape(var_shape, 1, &var_subshape)); in ScatterUpdateShape()
Ddata_flow_ops.cc77 TF_RETURN_IF_ERROR(c->Subshape(data_shape, rank, &data_suffix_shape)); in __anon4a7ac08d0202()
118 TF_RETURN_IF_ERROR(c->Subshape(data_shape, indices_rank, &rest)); in DynamicStitchShapeFunction()
768 TF_RETURN_IF_ERROR(c->Subshape(value_shape, 1, &fed_shape)); in __anon4a7ac08d1702()
Dnn_ops.cc1226 TF_RETURN_IF_ERROR(c->Subshape(input, 0, -1, &s)); in TopKShapeFn()
1282 TF_RETURN_IF_ERROR(c->Subshape(input, 0, -1, &s)); in __anon3e672dd82602()
/external/tensorflow/tensorflow/contrib/resampler/ops/
Dresampler_ops.cc36 TF_RETURN_IF_ERROR(c->Subshape(warp, 0, -1, &output)); in __anon447735090102()
/external/tensorflow/tensorflow/core/framework/
Dcommon_shape_fns.cc265 TF_RETURN_IF_ERROR(c->Subshape(input_shape, 0, 1, &first)); in BiasAddShape()
268 TF_RETURN_IF_ERROR(c->Subshape(input_shape, 2, &last)); in BiasAddShape()
280 TF_RETURN_IF_ERROR(c->Subshape(input_shape, 0, -1, &all_but_bias)); in BiasAddShape()
1260 TF_RETURN_IF_ERROR(c->Subshape(input, 0, concat_dim, &output_before)); in ConcatShapeHelper()
1265 TF_RETURN_IF_ERROR(c->Subshape(input, concat_dim + 1, &output_after)); in ConcatShapeHelper()
1273 TF_RETURN_IF_ERROR(c->Subshape(input, 0, concat_dim, &before)); in ConcatShapeHelper()
1278 TF_RETURN_IF_ERROR(c->Subshape(input, concat_dim + 1, &after)); in ConcatShapeHelper()
1567 c->Subshape(indices_shape, 0, num_outer_dims, &prefix_indices)); in ScatterNdUpdateShape()
1570 c->Subshape(updates_shape, 0, num_outer_dims, &prefix_updates)); in ScatterNdUpdateShape()
1583 TF_RETURN_IF_ERROR(c->Subshape(input_shape, ix, &input_suffix)); in ScatterNdUpdateShape()
[all …]
Dshape_inference_test.cc809 TEST_F(ShapeInferenceTest, Subshape) { in TEST_F() argument
816 EXPECT_TRUE(c.Subshape(unknown, 0, &out).ok()); in TEST_F()
819 EXPECT_TRUE(c.Subshape(unknown, 1, &out).ok()); in TEST_F()
822 EXPECT_TRUE(c.Subshape(unknown, 200, &out).ok()); in TEST_F()
829 EXPECT_TRUE(c.Subshape(in0, 0, &out).ok()); in TEST_F()
840 ASSERT_TRUE(c.Subshape(in0, start, end, &out_arr[0]).ok()); in TEST_F()
841 ASSERT_TRUE(c.Subshape(in0, neg_start, end, &out_arr[1]).ok()); in TEST_F()
842 ASSERT_TRUE(c.Subshape(in0, start, neg_end, &out_arr[2]).ok()); in TEST_F()
843 ASSERT_TRUE(c.Subshape(in0, neg_start, neg_end, &out_arr[3]).ok()); in TEST_F()
863 c.Subshape(in0, 6, -3, &out).ToString(), in TEST_F()
[all …]
Dshape_inference.h430 Status Subshape(ShapeHandle s, int64 start,
436 Status Subshape(ShapeHandle s, int64 start, int64 end,
443 Status Subshape(ShapeHandle s, int64 start, int64 end, int64 stride,
Dshape_inference.cc626 Status InferenceContext::Subshape(ShapeHandle s, int64 start, in Subshape() function in tensorflow::shape_inference::InferenceContext
628 return Subshape(s, start, std::numeric_limits<int64>::max() /* end */, out); in Subshape()
631 Status InferenceContext::Subshape(ShapeHandle s, int64 start, int64 end, in Subshape() function in tensorflow::shape_inference::InferenceContext
633 return Subshape(s, start, end, 1 /* stride */, out); in Subshape()
636 Status InferenceContext::Subshape(ShapeHandle s, int64 start, int64 end, in Subshape() function in tensorflow::shape_inference::InferenceContext
/external/tensorflow/tensorflow/core/grappler/costs/
Dgraph_properties.cc1497 TF_RETURN_IF_ERROR(ic->Subshape(input, start, &result)); in MaybeUpdateNodeContextOutput()
1500 TF_RETURN_IF_ERROR(ic->Subshape(input, start, end, &result)); in MaybeUpdateNodeContextOutput()
1555 TF_RETURN_IF_ERROR(ic->Subshape(input, begin, end, stride, &result)); in MaybeUpdateNodeContextOutput()
/external/tensorflow/tensorflow/core/common_runtime/
Dshape_refiner.cc617 TF_RETURN_IF_ERROR(ctx->Subshape(input, begin, end, stride, result)); in PartialStridedSliceShape()