Searched defs:reshape_dims (Results 1 – 15 of 15) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/image/ |
| D | adjust_contrast_op.h | 51 reshape_dims; in operator() local
99 reshape_dims; in operator() 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
|
| D | cxx11_tensor_padding_sycl.cpp | 107 Eigen::DSizes<IndexType, 2> reshape_dims; in test_padded_expr() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | cwise_op_gpu_select.cu.cc | 90 Eigen::IndexList<int, Eigen::type2index<1> > reshape_dims; in operator ()() local
|
| D | cwise_op_select.cc | 363 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/compiler/xla/service/gpu/ |
| D | gpu_conv_rewriter.cc | 515 std::vector<int64_t> reshape_dims = SpanToVector(rhs->shape().dimensions()); in MatchBackwardInput() local
623 std::vector<int64_t> reshape_dims = SpanToVector(lhs->shape().dimensions()); in ConvertBatchGroupedToFeatureGroupedConvolution() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | resampler_ops.cc | 267 std::vector<int64_t> 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 | 230 std::vector<int64_t> reshape_dims(original_shape.dimensions().begin(), in CombineShapePair() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
| D | legalize_common.cc | 1211 SmallVector<int64_t> reshape_dims; in convertExpandDimsOp() local
1265 SmallVector<int64_t> reshape_dims; in convertSqueezeOp() local
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.cc | 1447 nvinfer1::Dims reshape_dims; in TransposeTensor() local
|
| D | convert_nodes_test.cc | 729 const std::vector<int>& input_dims, const std::vector<int>& reshape_dims, in TestPrepareTensorForShape()
|