| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | diag_op.cc | 105 std::vector<int64> new_dims(dims.size() * 2); in Compile() local
137 std::vector<int64> new_dims; in Compile() local
|
| D | tensor_list_ops.cc | 495 std::vector<int64> new_dims = {num_elements * tensor_lengths}; in Compile() local
551 std::vector<int64> new_dims = {element_dims[0] / length, length}; in Compile() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | transpose_util_test.cc | 30 internal::TransposeDimsVec new_dims; in TestDimensionReduction() local
|
| D | transpose_functor.h | 81 TransposeDimsVec* new_dims) { in ReduceTransposeDimensions()
|
| D | transpose_functor_gpu.cu.cc | 102 TransposeDimsVec new_dims; in run() local
|
| /external/tensorflow/tensorflow/c/ |
| D | tf_tensor.cc | 171 TF_Tensor* to, const int64_t* new_dims, in TF_TensorBitcastFrom()
220 const int64_t* new_dims, int num_new_dims) { in BitcastFrom()
|
| D | c_api_test.cc | 1523 int64_t new_dims[] = {3, 2}; in TEST() local
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | resolve_constant_unary.cc | 85 std::vector<int> new_dims; in ReduceGeneric() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | dot_handler.cc | 781 std::vector<int64> new_dims; in PartitionBaseCase() local
939 std::vector<int64> new_dims(new_dot_shape.dimensions().begin(), in PartitionBaseCase() local
962 std::vector<int64> new_dims(dot_rhs->shape().dimensions().begin(), in PartitionBaseCase() local
970 std::vector<int64> new_dims(dot_lhs->shape().dimensions().begin(), in PartitionBaseCase() local
|
| D | spmd_partitioner.cc | 2245 std::vector<int64> new_dims; in HandleBroadcast() local
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | matrix.cc | 525 std::vector<int64> new_dims; in Einsum() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_op_kernel.cc | 156 int index, absl::Span<const int64> new_dims, in ConstantInputReshaped()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | indexed_array_analysis.cc | 458 DimensionVector new_dims; in StripDegenerateDimensions() local
|
| D | algebraic_simplifier.cc | 3156 std::vector<int64> new_dims; in HandleBroadcast() local
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.cc | 2669 nvinfer1::Dims new_dims; in ConvertExpandDims() local
2777 nvinfer1::Dims new_dims; in SqueezeTensor() local
3043 nvinfer1::Dims new_dims; in ConvertStridedSliceHelper() local
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ir/ |
| D | tfl_ops.cc | 343 int new_dims) { in GetPaddedShape()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | constant_folding.cc | 1705 PartialTensorShape new_dims; in IsSimplifiableReshape() local
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/IR/ |
| D | hlo_ops.cc | 1730 llvm::SmallVector<APInt, 5> new_dims; in fold() local
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 1382 llvm::SmallVector<int64_t, 4> new_dims; in matchAndRewrite() local
|