Searched defs:reshaped (Results 1 – 15 of 15) sorted by relevance
| /external/eigen/Eigen/src/plugins/ |
| D | ReshapedMethods.h | 93 reshaped(NRowsType nRows, NColsType nCols) EIGEN_RESHAPED_METHOD_CONST in reshaped() function
109 reshaped(NRowsType nRows, NColsType nCols) EIGEN_RESHAPED_METHOD_CONST in reshaped() function
124 reshaped() EIGEN_RESHAPED_METHOD_CONST in reshaped() function
133 reshaped() EIGEN_RESHAPED_METHOD_CONST in reshaped() function
|
| /external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/mhlo/transforms/ |
| D | optimize_mhlo.cc | 123 auto reshaped = rewriter.create<ReshapeOp>( in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | depthtospace_op.cc | 146 xla::XlaOp reshaped = xla::Reshape(input, reshaped_shape); in Compile() local
|
| D | spacetodepth_op.cc | 151 xla::XlaOp reshaped = xla::Reshape(input, reshaped_shape); in Compile() local
|
| D | batchtospace_op.cc | 75 xla::XlaOp reshaped = xla::Reshape(input, reshaped_shape); in BatchToSpace() local
|
| D | where_op.cc | 170 XlaOp reshaped = xla::Reshape(iota, {flattened_size}); in CompileWhereWithSort() local
|
| D | tensor_list_ops.cc | 564 xla::XlaOp reshaped = xla::Reshape(input_tensor, new_dims); in Compile() local
|
| /external/tensorflow/tensorflow/core/grappler/clusters/ |
| D | single_machine_test.cc | 177 auto reshaped = ops::Reshape(root.WithOpName("reshaped"), square, new_shape); in TEST_F() local
|
| /external/eigen/unsupported/test/ |
| D | cxx11_tensor_morphing.cpp | 54 Tensor<float, 3> reshaped = tensor.reshape(static_cast<Eigen::DSizes<ptrdiff_t,3>>(dim)); in test_static_reshape() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | lower_tf.cc | 1003 auto reshaped = rewriter.create<ReshapeOp>( in matchAndRewrite() local
1116 auto reshaped = rewriter.create<TF::ReshapeOp>( in matchAndRewrite() local
|
| D | legalize_hlo.cc | 1758 auto reshaped = in ConvertDot() local
|
| /external/tensorflow/tensorflow/cc/gradients/ |
| D | nn_grad.cc | 72 auto reshaped = ExpandDims(scope, vec, -1); in BroadcastMul() local
|
| D | math_grad.cc | 947 auto reshaped = Reshape( in ProdGrad() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | spmd_partitioner.cc | 1790 PartitionedHlo reshaped = SplitReshapeHelper( in TryComplexReshardHandling() local
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 2160 Value reshaped = op.input(); in matchAndRewrite() local
|