Searched defs:reshaped (Results 1 – 15 of 15) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | where_op.cc | 65 xla::XlaOp reshaped = xla::Reshape(iota, {flattened_size}); in Compile() local
|
| 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 | tensor_list_ops.cc | 556 xla::XlaOp reshaped = xla::Reshape(input_tensor, new_dims); in Compile() local
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | optimize_mhlo.cc | 136 auto reshaped = rewriter.create<ReshapeOp>( in matchAndRewrite() local
|
| D | transform_unranked_hlo.cc | 185 Value reshaped = rewriter.create<mhlo::DynamicReshapeOp>( in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/vectorization/ |
| D | cwise_op_vectorizer.cc | 103 Output reshaped = ops::Reshape(scope, input, new_shape); in ExpandDimsForBroadcast() 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/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | lower_tf.cc | 973 auto reshaped = rewriter.create<ReshapeOp>( in matchAndRewrite() local
1087 auto reshaped = rewriter.create<TF::ReshapeOp>( in matchAndRewrite() local
|
| D | legalize_hlo.cc | 535 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 | 992 auto reshaped = Reshape( in ProdGrad() local
|
| /external/llvm-project/mlir/test/EDSC/ |
| D | builder-api-test.cpp | 1062 auto reshaped = linalg_reshape(v, maps); in TEST_FUNC() local
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 1761 Value reshaped = op.input(); in matchAndRewrite() local
|