Home
last modified time | relevance | path

Searched refs:Subshape (Results 1 – 20 of 20) 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()
99 TF_RETURN_IF_ERROR(c->Subshape(lhs, 0, -2, &lhs_batch_shape)); in MatrixTriangularSolveShapeFn()
100 TF_RETURN_IF_ERROR(c->Subshape(rhs, 0, -2, &rhs_batch_shape)); in MatrixTriangularSolveShapeFn()
124 TF_RETURN_IF_ERROR(c->Subshape(input, 0, -2, &batch_shape)); in SelfAdjointEigV2ShapeFn()
151 TF_RETURN_IF_ERROR(c->Subshape(input, 0, -2, &batch_shape)); in LuShapeFn()
177 TF_RETURN_IF_ERROR(c->Subshape(input, 0, -2, &batch_shape)); in QrShapeFn()
208 TF_RETURN_IF_ERROR(c->Subshape(input, 0, -2, &batch_shape)); in SvdShapeFn()
258 TF_RETURN_IF_ERROR(c->Subshape(superdiag, 0, -2, &superdiag_batch_shape)); in TridiagonalMatMulShapeFn()
[all …]
Dresource_variable_ops.cc265 TF_RETURN_IF_ERROR(c->Subshape(handle_shape_and_type[0].shape, 0, in __anond0909b4b0402()
269 TF_RETURN_IF_ERROR(c->Subshape(handle_shape_and_type[0].shape, in __anond0909b4b0402()
274 c->Subshape(indices_shape, batch_dims, &indices_subshape)); in __anond0909b4b0402()
315 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 __anond96a6a670702()
281 TF_RETURN_IF_ERROR(c->Subshape(b_shape, 0, -2, &b_batch_dims)); in __anond96a6a670702()
433 TF_RETURN_IF_ERROR(c->Subshape(a_shape, 0, -2, &a_batch_dims)); in __anond96a6a670a02()
434 TF_RETURN_IF_ERROR(c->Subshape(b_shape, 0, -2, &b_batch_dims)); in __anond96a6a670a02()
598 TF_RETURN_IF_ERROR(c->Subshape(matrix_shape, 0, -2, &matrix_batch_shape)); in __anond96a6a671002()
599 TF_RETURN_IF_ERROR(c->Subshape(perm_shape, 0, -1, &perm_shape)); in __anond96a6a671002()
Dset_ops.cc66 c->Subshape(input0_shape, 0, rank - 1, &group0_shape)); in __anon0af7a88a0102()
69 c->Subshape(input1_shape, 0, rank - 1, &group1_shape)); in __anon0af7a88a0102()
Dencode_proto_ops.cc41 TF_RETURN_IF_ERROR(c->Subshape(input, 0, -1, &inner)); in __anond4c3afbf0102()
Dragged_array_ops.cc79 TF_RETURN_IF_ERROR(c->Subshape(params_dense_values, 1, &value)); in RaggedGatherShapeFn()
Dcollective_ops.cc53 TF_RETURN_IF_ERROR(c->Subshape(c->input(0), 1, &in_subshape)); in __anonea321d280102()
Darray_ops.cc880 TF_RETURN_IF_ERROR(c->Subshape(diag, 0, -1, &diag_batch_shape)); in __anonf6523ebd1102()
1167 TF_RETURN_IF_ERROR(c->Subshape(c->input(0), 1, &params_subshape)); in __anonf6523ebd1702()
1228 c->Subshape(params_shape, 0, axis, &params_outer_subshape)); in __anonf6523ebd1802()
1232 c->Subshape(indices_shape, batch_dims, &indices_inner_subshape)); in __anonf6523ebd1802()
1245 c->Subshape(params_shape, axis + 1, &params_inner_subshape)); in __anonf6523ebd1802()
2063 TF_RETURN_IF_ERROR(c->Subshape(input, dim, &end)); in __anonf6523ebd2e02()
2067 TF_RETURN_IF_ERROR(c->Subshape(input, 0, dim, &output)); in __anonf6523ebd2e02()
2248 c->Subshape(input_shape, 1 + num_block_dims, &remaining_input_shape)); in SpaceToBatchShapeHelper()
2323 c->Subshape(input_shape, 1 + num_block_dims, &remaining_input_shape)); in BatchToSpaceShapeHelper()
2749 TF_RETURN_IF_ERROR(c->Subshape(indices, 0, depth_index, &front)); in __anonf6523ebd3a02()
[all …]
Dlist_ops.cc270 TF_RETURN_IF_ERROR(c->Subshape(element_shape, 1, &result)); in TensorListConcatShapeInference()
330 c->Subshape(tensor_shape, 1, &element_shape_from_tensor_shape)); in __anonef6dc5da0902()
359 c->Subshape(tensor_shape, 1, &tensor_shape_except_first_dim)); in __anonef6dc5da0a02()
Dstate_ops.cc123 TF_RETURN_IF_ERROR(c->Subshape(var_shape, 1, &var_subshape)); in ScatterUpdateShape()
Dmath_ops.cc1112 TF_RETURN_IF_ERROR(c->Subshape(data_shape, 1, &subshape)); in SegmentReductionShapeFn()
1136 TF_RETURN_IF_ERROR(c->Subshape(data_shape, 1, &subshape)); in SparseSegmentReductionShapeFn()
1160 TF_RETURN_IF_ERROR(c->Subshape(data_shape, 1, &subshape)); in SparseSegmentReductionGradShapeFn()
1201 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 __anonb84a3fcf0202()
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 __anonb84a3fcf1b02()
Dnn_ops.cc1291 TF_RETURN_IF_ERROR(c->Subshape(input, 0, -1, &s)); in TopKShapeFn()
1347 TF_RETURN_IF_ERROR(c->Subshape(input, 0, -1, &s)); in __anon87c0335a2602()
/external/tensorflow/tensorflow/core/framework/
Dcommon_shape_fns.cc335 TF_RETURN_IF_ERROR(c->Subshape(input_shape, axis_before_ellipsis, in EinsumShape()
441 TF_RETURN_IF_ERROR(c->Subshape(a_shape, 0, -2, &a_batch_shape)); in BatchMatMulV2Shape()
442 TF_RETURN_IF_ERROR(c->Subshape(b_shape, 0, -2, &b_batch_shape)); in BatchMatMulV2Shape()
473 TF_RETURN_IF_ERROR(c->Subshape(a_shape, 0, -2, &a_batch_dims)); in BatchMatMulShape()
474 TF_RETURN_IF_ERROR(c->Subshape(b_shape, 0, -2, &b_batch_dims)); in BatchMatMulShape()
520 TF_RETURN_IF_ERROR(c->Subshape(input_shape, 0, 1, &first)); in BiasAddShape()
523 TF_RETURN_IF_ERROR(c->Subshape(input_shape, 2, &last)); in BiasAddShape()
535 TF_RETURN_IF_ERROR(c->Subshape(input_shape, 0, -1, &all_but_bias)); in BiasAddShape()
1426 TF_RETURN_IF_ERROR(c->Subshape( in MatrixSetDiagV2Shape()
1836 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.h395 Status Subshape(ShapeHandle s, int64 start,
401 Status Subshape(ShapeHandle s, int64 start, int64 end,
408 Status Subshape(ShapeHandle s, int64 start, int64 end, int64 stride,
Dshape_inference.cc566 Status InferenceContext::Subshape(ShapeHandle s, int64 start, in Subshape() function in tensorflow::shape_inference::InferenceContext
568 return Subshape(s, start, std::numeric_limits<int64>::max() /* end */, out); in Subshape()
571 Status InferenceContext::Subshape(ShapeHandle s, int64 start, int64 end, in Subshape() function in tensorflow::shape_inference::InferenceContext
573 return Subshape(s, start, end, 1 /* stride */, out); in Subshape()
576 Status InferenceContext::Subshape(ShapeHandle s, int64 start, int64 end, in Subshape() function in tensorflow::shape_inference::InferenceContext
/external/tensorflow/tensorflow/c/
Dops.cc257 Status s = cc_ctx->Subshape(*reinterpret_cast<ShapeHandle*>(shape_handle), in TF_ShapeInferenceContextSubshape()
/external/tensorflow/tensorflow/core/common_runtime/
Dshape_refiner.cc635 TF_RETURN_IF_ERROR(ctx->Subshape(input, begin, end, stride, result)); in PartialStridedSliceShape()
/external/tensorflow/tensorflow/core/grappler/costs/
Dgraph_properties.cc1607 TF_RETURN_IF_ERROR(ic->Subshape(input, start, &result)); in MaybeUpdateNodeContextOutput()
1610 TF_RETURN_IF_ERROR(ic->Subshape(input, start, end, &result)); in MaybeUpdateNodeContextOutput()
1665 TF_RETURN_IF_ERROR(ic->Subshape(input, begin, end, stride, &result)); in MaybeUpdateNodeContextOutput()