| /external/tensorflow/tensorflow/core/ops/ |
| D | batch_ops.cc | 63 c->ReplaceDim(in_shapes[i], 0, c->UnknownDim(), &out_shapes[i])); in __anon8ccc74200102()
86 c->ReplaceDim(c->input(0), 0, c->UnknownDim(), &out_shape)); in __anon8ccc74200202()
|
| D | spectral_ops.cc | 94 TF_RETURN_IF_ERROR(c->ReplaceDim(out, -rank + i, c->UnknownDim(), &out)); in RFFTShape()
103 TF_RETURN_IF_ERROR(c->ReplaceDim(out, -rank + i, c->MakeDim(dim), &out)); in RFFTShape()
|
| D | ctc_ops.cc | 58 TF_RETURN_IF_ERROR(c->ReplaceDim(inputs, 1, batch_size, &inputs)); in __anoned3a12980102()
95 TF_RETURN_IF_ERROR(c->ReplaceDim(inputs, 1, batch_size, &inputs)); in __anoned3a12980202()
|
| D | image_ops.cc | 105 TF_RETURN_IF_ERROR(c->ReplaceDim(input, -1, last_dim, &out)); in ColorspaceShapeFn()
362 TF_RETURN_IF_ERROR(c->ReplaceDim(input, 1, c->UnknownDim(), &input)); in __anone43000c10602()
363 TF_RETURN_IF_ERROR(c->ReplaceDim(input, 2, c->UnknownDim(), &input)); in __anone43000c10602()
367 c->ReplaceDim(input, 1, c->MakeDim(size_vec(0)), &input)); in __anone43000c10602()
369 c->ReplaceDim(input, 2, c->MakeDim(size_vec(1)), &input)); in __anone43000c10602()
|
| D | nn_ops.cc | 122 TF_RETURN_IF_ERROR(c->ReplaceDim(input, 3, last_dim, &out)); in __anon87c0335a0302()
156 TF_RETURN_IF_ERROR(c->ReplaceDim(input, 3, last_dim, &dx)); in __anon87c0335a0402()
431 TF_RETURN_IF_ERROR(c->ReplaceDim(resized, 1, new_height, &resized)); in CommonFusedConvCalculations()
432 TF_RETURN_IF_ERROR(c->ReplaceDim(resized, 2, new_width, &resized)); in CommonFusedConvCalculations()
1216 TF_RETURN_IF_ERROR(c->ReplaceDim(features, 0, batch_size, &features)); in __anon87c0335a2202()
1605 TF_RETURN_IF_ERROR(c->ReplaceDim(input, 3, last_dim, &out)); in __anon87c0335a3002()
2545 TF_RETURN_IF_ERROR(c->ReplaceDim(x, 3, channel_dim, &y)); in __anon87c0335a4002()
2547 TF_RETURN_IF_ERROR(c->ReplaceDim(x, 1, channel_dim, &y)); in __anon87c0335a4002()
2619 c->ReplaceDim(y_backprop, 3, channel_dim, &x_backprop)); in __anon87c0335a4102()
2622 c->ReplaceDim(y_backprop, 1, channel_dim, &x_backprop)); in __anon87c0335a4102()
|
| D | control_flow_ops.cc | 136 TF_RETURN_IF_ERROR(c->ReplaceDim(out, d, c->UnknownDim(), &out)); in MergeShape()
|
| D | sparse_csr_matrix_ops.cc | 331 TF_RETURN_IF_ERROR(c->ReplaceDim(b_shape, 0, batch_size, &b_shape)); in __anond96a6a670802()
332 TF_RETURN_IF_ERROR(c->ReplaceDim(a_shape, 0, batch_size, &a_shape)); in __anond96a6a670802()
|
| D | array_ops.cc | 255 c->ReplaceDim(out, out_unknown_idx, inferred_dim, &out)); in SetOutputShapeForReshape()
273 c->ReplaceDim(out, out_unknown_idx, unknown_in_dim, &out)); in SetOutputShapeForReshape()
301 TF_RETURN_IF_ERROR(c->ReplaceDim( in __anonf6523ebd0202()
503 c->ReplaceDim(out, out_rank - i - 1, out_dim, &out)); in __anonf6523ebd0702()
595 c->ReplaceDim(input, split_dim, split_dim_size, &out)); in __anonf6523ebd0a02()
632 TF_RETURN_IF_ERROR(c->ReplaceDim(output_shape, in __anonf6523ebd0b02()
684 c->ReplaceDim(input, split_dim, c->MakeDim(size), &output_shape)); in __anonf6523ebd0b02()
1596 c->ReplaceDim(input, batch_dim, batch_dim_dim, &output_shape)); in __anonf6523ebd2302()
|
| D | lookup_ops.cc | 136 c->ReplaceDim(keys, keys_rank - key_suffix_rank + d, dim, &keys)); in ValidateTableResourceHandle()
|
| D | training_ops.cc | 67 c->ReplaceDim(grad, 0, c->UnknownDim(), &grad_unknown_first)); in HandleGradAndIndicesInputs()
|
| D | math_ops.cc | 1805 c->ReplaceDim(output, rank - 1, inferred_dim, &output)); in __anonb5dc68881402()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | shape_inference_test.cc | 923 TEST_F(ShapeInferenceTest, ReplaceDim) { in TEST_F() argument
932 EXPECT_TRUE(c.ReplaceDim(in, 0, c.Dim(in, 1), &replaced).ok()); in TEST_F()
934 EXPECT_TRUE(c.ReplaceDim(in, 2, c.Dim(in, 1), &replaced).ok()); in TEST_F()
936 EXPECT_TRUE(c.ReplaceDim(in, 1, c.Dim(in, 2), &replaced).ok()); in TEST_F()
938 EXPECT_TRUE(c.ReplaceDim(unknown, 0, c.Dim(in, 1), &replaced).ok()); in TEST_F()
942 EXPECT_TRUE(c.ReplaceDim(in, -1, c.Dim(in, 1), &replaced).ok()); in TEST_F()
944 EXPECT_TRUE(c.ReplaceDim(unknown, -1, c.Dim(in, 1), &replaced).ok()); in TEST_F()
948 EXPECT_FALSE(c.ReplaceDim(in, 3, c.Dim(in, 1), &replaced).ok()); in TEST_F()
951 EXPECT_FALSE(c.ReplaceDim(in, -4, c.Dim(in, 1), &replaced).ok()); in TEST_F()
|
| D | common_shape_fns.cc | 1105 TF_RETURN_IF_ERROR(c->ReplaceDim(x, channel_dim_index, channel_dim, &y)); in FusedBatchNormShape()
1175 c->ReplaceDim(y_backprop, channel_dim_index, channel_dim, &x_backprop)); in FusedBatchNormGradShape()
1355 TF_RETURN_IF_ERROR(c->ReplaceDim(input_shape, input_rank - 1, in MatrixDiagV2Shape()
1360 TF_RETURN_IF_ERROR(c->ReplaceDim(input_shape, input_rank - 2, in MatrixDiagV2Shape()
1362 TF_RETURN_IF_ERROR(c->ReplaceDim(output_shape, input_rank - 1, in MatrixDiagV2Shape()
|
| D | shape_inference.h | 418 Status ReplaceDim(ShapeHandle s, int64 dim_index, DimensionHandle new_dim,
|
| D | shape_inference.cc | 651 Status InferenceContext::ReplaceDim(ShapeHandle s, int64 dim_index_in, in ReplaceDim() function in tensorflow::shape_inference::InferenceContext
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | graph_properties.cc | 996 TF_CHECK_OK(ctx->ReplaceDim(relaxed, d, new_dim, &relaxed)); in OutputAsUnion()
|