| /external/tensorflow/tensorflow/core/ops/ |
| D | batch_ops.cc | 63 c->ReplaceDim(in_shapes[i], 0, c->UnknownDim(), &out_shapes[i])); in __anon130e45de0102()
86 c->ReplaceDim(c->input(0), 0, c->UnknownDim(), &out_shape)); in __anon130e45de0202()
|
| 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 | image_ops.cc | 105 TF_RETURN_IF_ERROR(c->ReplaceDim(input, -1, last_dim, &out)); in ColorspaceShapeFn()
336 TF_RETURN_IF_ERROR(c->ReplaceDim(input, 1, c->UnknownDim(), &input)); in __anon6a71d27f0602()
337 TF_RETURN_IF_ERROR(c->ReplaceDim(input, 2, c->UnknownDim(), &input)); in __anon6a71d27f0602()
341 c->ReplaceDim(input, 1, c->MakeDim(size_vec(0)), &input)); in __anon6a71d27f0602()
343 c->ReplaceDim(input, 2, c->MakeDim(size_vec(1)), &input)); in __anon6a71d27f0602()
|
| D | nn_ops.cc | 121 TF_RETURN_IF_ERROR(c->ReplaceDim(input, 3, last_dim, &out)); in __anon3e672dd80302()
155 TF_RETURN_IF_ERROR(c->ReplaceDim(input, 3, last_dim, &dx)); in __anon3e672dd80402()
366 TF_RETURN_IF_ERROR(c->ReplaceDim(resized, 1, new_height, &resized)); in CommonFusedConvCalculations()
367 TF_RETURN_IF_ERROR(c->ReplaceDim(resized, 2, new_width, &resized)); in CommonFusedConvCalculations()
1151 TF_RETURN_IF_ERROR(c->ReplaceDim(features, 0, batch_size, &features)); in __anon3e672dd82202()
1540 TF_RETURN_IF_ERROR(c->ReplaceDim(input, 3, last_dim, &out)); in __anon3e672dd83002()
2407 TF_RETURN_IF_ERROR(c->ReplaceDim(x, 3, channel_dim, &y)); in __anon3e672dd83e02()
2409 TF_RETURN_IF_ERROR(c->ReplaceDim(x, 1, channel_dim, &y)); in __anon3e672dd83e02()
2481 c->ReplaceDim(y_backprop, 3, channel_dim, &x_backprop)); in __anon3e672dd83f02()
2484 c->ReplaceDim(y_backprop, 1, channel_dim, &x_backprop)); in __anon3e672dd83f02()
|
| D | ctc_ops.cc | 57 TF_RETURN_IF_ERROR(c->ReplaceDim(inputs, 1, batch_size, &inputs)); in __anon78c05cd60102()
|
| D | control_flow_ops.cc | 106 TF_RETURN_IF_ERROR(c->ReplaceDim(out, d, c->UnknownDim(), &out)); in MergeShape()
|
| D | array_ops.cc | 249 c->ReplaceDim(out, out_unknown_idx, inferred_dim, &out)); in SetOutputShapeForReshape()
267 c->ReplaceDim(out, out_unknown_idx, unknown_in_dim, &out)); in SetOutputShapeForReshape()
295 TF_RETURN_IF_ERROR(c->ReplaceDim( in __anon7c94107b0202()
497 c->ReplaceDim(out, out_rank - i - 1, out_dim, &out)); in __anon7c94107b0702()
589 c->ReplaceDim(input, split_dim, split_dim_size, &out)); in __anon7c94107b0a02()
626 TF_RETURN_IF_ERROR(c->ReplaceDim(output_shape, in __anon7c94107b0b02()
678 c->ReplaceDim(input, split_dim, c->MakeDim(size), &output_shape)); in __anon7c94107b0b02()
1507 c->ReplaceDim(input, batch_dim, batch_dim_dim, &output_shape)); in __anon7c94107b2402()
|
| D | lookup_ops.cc | 136 c->ReplaceDim(keys, keys_rank - key_suffix_rank + d, dim, &keys)); in ValidateTableResourceHandle()
|
| D | training_ops.cc | 53 c->ReplaceDim(grad, 0, c->UnknownDim(), &grad_unknown_first)); in HandleGradAndIndicesInputs()
|
| D | math_ops.cc | 1702 c->ReplaceDim(output, rank - 1, inferred_dim, &output)); in __anonb22bfa861402()
|
| /external/tensorflow/tensorflow/contrib/coder/ops/ |
| D | coder_ops.cc | 130 TF_RETURN_IF_ERROR(c->ReplaceDim(in, -1, last, &out)); in __anon8696facd0202()
|
| /external/tensorflow/tensorflow/contrib/mpi_collectives/ops/ |
| D | mpi_ops.cc | 109 c->ReplaceDim(c->input(0), 0, c->UnknownDim(), &output)); in __anon107d6f1b0502()
|
| /external/tensorflow/tensorflow/contrib/reduce_slice_ops/ops/ |
| D | reduce_slice_ops.cc | 58 TF_RETURN_IF_ERROR(c->ReplaceDim(handle, axis, dim_axis, &handle)); in ReduceSliceShapeFn()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | shape_inference_test.cc | 908 TEST_F(ShapeInferenceTest, ReplaceDim) { in TEST_F() argument
917 EXPECT_TRUE(c.ReplaceDim(in, 0, c.Dim(in, 1), &replaced).ok()); in TEST_F()
919 EXPECT_TRUE(c.ReplaceDim(in, 2, c.Dim(in, 1), &replaced).ok()); in TEST_F()
921 EXPECT_TRUE(c.ReplaceDim(in, 1, c.Dim(in, 2), &replaced).ok()); in TEST_F()
923 EXPECT_TRUE(c.ReplaceDim(unknown, 0, c.Dim(in, 1), &replaced).ok()); in TEST_F()
927 EXPECT_TRUE(c.ReplaceDim(in, -1, c.Dim(in, 1), &replaced).ok()); in TEST_F()
929 EXPECT_TRUE(c.ReplaceDim(unknown, -1, c.Dim(in, 1), &replaced).ok()); in TEST_F()
933 EXPECT_FALSE(c.ReplaceDim(in, 3, c.Dim(in, 1), &replaced).ok()); in TEST_F()
936 EXPECT_FALSE(c.ReplaceDim(in, -4, c.Dim(in, 1), &replaced).ok()); in TEST_F()
|
| D | shape_inference.h | 453 Status ReplaceDim(ShapeHandle s, int64 dim_index, DimensionHandle new_dim,
|
| D | common_shape_fns.cc | 814 TF_RETURN_IF_ERROR(c->ReplaceDim(x, channel_dim_index, channel_dim, &y)); in FusedBatchNormShape()
853 c->ReplaceDim(y_backprop, channel_dim_index, channel_dim, &x_backprop)); in FusedBatchNormGradShape()
|
| D | shape_inference.cc | 711 Status InferenceContext::ReplaceDim(ShapeHandle s, int64 dim_index_in, in ReplaceDim() function in tensorflow::shape_inference::InferenceContext
|
| /external/tensorflow/tensorflow/contrib/mpi_collectives/ |
| D | mpi_ops.cc | 1204 c->ReplaceDim(c->input(0), 0, c->UnknownDim(), &output)); in __anon976d40da0a02()
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | graph_properties.cc | 947 TF_CHECK_OK(ctx->ReplaceDim(relaxed, d, new_dim, &relaxed)); in OutputAsUnion()
|