| /external/tensorflow/tensorflow/core/kernels/ |
| D | spacetobatch_benchmark_test.cc | 88 BM_SpaceToBatch(BatchToSpace, 256, 50, 50, 64, 2, 0, 0, 0, 0);
89 BM_SpaceToBatch(BatchToSpace, 256, 50, 50, 1, 2, 0, 0, 0, 0);
90 BM_SpaceToBatch(BatchToSpace, 256, 50, 50, 64, 2, 3, 3, 3, 3);
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | batchtospace_op.cc | 24 void BatchToSpace(XlaOpKernelContext* ctx, const xla::XlaOp& input, in BatchToSpace() function
157 BatchToSpace(ctx, ctx->Input(0), input_type(0), ctx->InputShape(0), in Compile()
179 BatchToSpace(ctx, ctx->Input(0), input_type(0), ctx->InputShape(0), in Compile()
|
| /external/tensorflow/tensorflow/core/api_def/java_api/ |
| D | api_def_BatchToSpace.pbtxt | 2 graph_op_name: "BatchToSpace"
|
| /external/tensorflow/tensorflow/core/api_def/python_api/ |
| D | api_def_BatchToSpace.pbtxt | 2 graph_op_name: "BatchToSpace"
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_BatchToSpace.pbtxt | 2 graph_op_name: "BatchToSpace"
94 summary: "BatchToSpace for 4-D tensors of type T."
|
| D | api_def_BatchToSpaceND.pbtxt | 130 summary: "BatchToSpace for N-D tensors of type T."
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | BatchToSpace.pbtxt | 2 name: "BatchToSpace"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | BatchToSpace.pbtxt | 2 name: "BatchToSpace"
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/ |
| D | tf-ops.mlir | 3782 // tf.BatchToSpace
3786 …%0 = "tf.BatchToSpace"(%arg0, %arg1) {block_size = 2 : i64} : (tensor<*xf32>, tensor<*xi32>) -> te…
3791 …%0 = "tf.BatchToSpace"(%arg0, %arg1) {block_size = 2 : i64} : (tensor<?x?x?x?xf32>, tensor<*xi32>)…
3796 …%0 = "tf.BatchToSpace"(%arg0, %arg1) {block_size = 2 : i64} : (tensor<*xf32>, tensor<?x?xi32>) -> …
3801 …%0 = "tf.BatchToSpace"(%arg0, %arg1) {block_size = 2 : i64} : (tensor<*xf32>, tensor<*xi32>) -> te…
3807 …%0 = "tf.BatchToSpace"(%arg0, %crops) {block_size = 3 : i64} : (tensor<36x8x8x8xf32>, tensor<2x2xi…
3814 …// expected-error @+1 {{'tf.BatchToSpace' op requires input to be a 4D tensor, but got 'tensor<8xf…
3815 …%0 = "tf.BatchToSpace"(%arg0, %arg1) {block_size = 2 : i64} : (tensor<8xf32>, tensor<*xi32>) -> te…
3822 …// expected-error @+1 {{'tf.BatchToSpace' op requires input batch (dimension 0) to be evenly divis…
3823 …%0 = "tf.BatchToSpace"(%arg0, %arg1) {block_size = 2 : i64} : (tensor<2x4x6x8xf32>, tensor<*xi32>)…
[all …]
|
| D | canonicalize.mlir | 1153 …%0 = "tf.BatchToSpace"(%arg0, %arg1) {block_size = 8 : i64} : (tensor<?x?x?x?xf32>, tensor<?x?xi32…
1161 …%0 = "tf.BatchToSpace"(%arg0, %arg1) {block_size = 8 : i64} : (tensor<*xf32>, tensor<?x?xi32>) -> …
1168 …%0 = "tf.BatchToSpace"(%arg0, %arg1) {block_size = 8 : i64} : (tensor<?x?x?x?xf32>, tensor<*xi32>)…
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | space_to_batch_converter.cc | 173 StatusOr<HloInstruction*> BatchToSpace(HloInstruction* old_instr);
660 BatchToSpace(instr->mutable_operand(i))); in Run()
1706 StatusOr<HloInstruction*> ConvolutionVisitor::BatchToSpace( in BatchToSpace() function in xla::__anona59094b40111::ConvolutionVisitor
1765 BatchToSpace(old_conv)); in PropagateOnUsers()
1806 TF_ASSIGN_OR_RETURN(HloInstruction * batch_to_space, BatchToSpace(node)); in PropagateOnUsers()
1823 BatchToSpace(node)); in PropagateOnUsers()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | canonicalize.td | 81 // BatchToSpace op patterns.
|
| /external/tensorflow/tensorflow/cc/gradients/ |
| D | array_grad_test.cc | 299 auto y = BatchToSpace(scope_, x, paddings, /* block_size */ 2); in TEST_F()
|
| D | array_grad.cc | 328 BatchToSpace(scope, grad_inputs[0], op.input(1), block_size)); in SpaceToBatchGrad()
|
| /external/tensorflow/tensorflow/lite/g3doc/guide/ |
| D | op_select_allowlist.md | 65 * `raw_ops.BatchToSpace`
|
| /external/tensorflow/tensorflow/compiler/tf2xla/g3doc/ |
| D | gpu_supported_ops.md | 32 `BatchToSpace` | `Tidx={int32,int64}`<br>`T={bool,complex64,double,float,int…
|
| D | cpu_supported_ops.md | 32 `BatchToSpace` | `Tidx={int32,int64}`<br>`T={bool,complex64,double,float,int…
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | randomized_tests.cc | 1229 TEST_F(OpTest, BatchToSpace) { in TEST_F() argument
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_generated_ops.td | 980 def TF_BatchToSpaceOp : TF_Op<"BatchToSpace", [NoSideEffect]> {
981 let summary = "BatchToSpace for 4-D tensors of type T.";
1088 let summary = "BatchToSpace for N-D tensors of type T.";
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.raw_ops.pbtxt | 428 name: "BatchToSpace"
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.raw_ops.pbtxt | 428 name: "BatchToSpace"
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v0.pbtxt | 6380 name: "BatchToSpace"
6405 name: "BatchToSpace"
|
| D | ops_history.v2.pbtxt | 10346 name: "BatchToSpace"
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 1337 func BatchToSpace(scope *Scope, input tf.Output, crops tf.Output, block_size int64) (output tf.Outp… func
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 4280 name: "BatchToSpace"
|