| /external/tensorflow/tensorflow/core/kernels/image/ |
| D | adjust_contrast_op.h | 52 Eigen::Tensor<int, 4>::Dimensions reshape_dims; in operator() local
65 reshape_dims; in operator() local
114 Eigen::Tensor<int, 4>::Dimensions reshape_dims; in operator() local
132 reshape_dims; in operator() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | cwise_op_gpu_select.cu.cc | 85 Eigen::Tensor<int, 2>::Dimensions reshape_dims{{ batch, 1 }}; in operator ()() local
89 Eigen::IndexList<int, Eigen::type2index<1> > reshape_dims; in operator ()() local
|
| D | cwise_op_select.cc | 342 Eigen::Tensor<Eigen::DenseIndex, 2>::Dimensions reshape_dims{{batch, 1}}; in operator ()() local
346 Eigen::IndexList<Eigen::DenseIndex, Eigen::type2index<1> > reshape_dims; in operator ()() local
|
| D | eigen_spatial_convolutions_test.cc | 1511 NewDimension reshape_dims; in PackRhsHelper() local
1649 NewDimension reshape_dims; in PackLhsHelper() local
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | convert_expanddims_to_reshape.cc | 58 std::vector<int> reshape_dims(input_array.shape().dims()); in Run() local
|
| D | convert_reorder_axes.cc | 40 std::vector<int> reshape_dims = {1, input_shape.dims(0), input_shape.dims(1), in CreateReshapeFromReorderAxes() local
|
| /external/eigen/unsupported/test/ |
| D | cxx11_tensor_padding.cpp | 63 Eigen::DSizes<ptrdiff_t, 2> reshape_dims; in test_padded_expr() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_conv_rewriter.cc | 87 std::vector<int64> reshape_dims = SpanToVector(lhs->shape().dimensions()); in ConvertBatchGroupedToFeatureGroupedConvolution() local
567 std::vector<int64> reshape_dims = SpanToVector(rhs->shape().dimensions()); in MatchBackwardInput() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | resampler_ops.cc | 264 std::vector<int64> reshape_dims(warp_shape.dims()); in CalculateGradData() local
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | quantize_nodes.cc | 747 NodeDef reshape_dims; in QuantizeNodes() local
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | prng.cc | 228 std::vector<int64> reshape_dims(original_shape.dimensions().begin(), in CombineShapePair() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
| D | legalize_common.cc | 1265 SmallVector<int64_t, 4> reshape_dims; in convertExpandDimsOp() local
1312 SmallVector<int64_t, 8> reshape_dims; in convertSqueezeOp() local
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.cc | 1616 nvinfer1::Dims reshape_dims; in TransposeTensor() local
2922 nvinfer1::Dims reshape_dims; in ConvertStridedSliceHelper() local
4113 nvinfer1::Dims reshape_dims; in ConvertBiasAddInt8WithoutCalibration() local
4151 nvinfer1::Dims reshape_dims = original_dims; in ConvertBiasAddInt8WithoutCalibration() local
|
| D | convert_nodes_test.cc | 937 const std::vector<int>& input_dims, const std::vector<int>& reshape_dims, in TestPrepareTensorForShape()
|