Searched refs:BatchToSpace (Results 1 – 18 of 18) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | spacetobatch_benchmark_test.cc | 83 BM_SpaceToBatch(BatchToSpace, 256, 50, 50, 64, 2, 0, 0, 0, 0);
84 BM_SpaceToBatch(BatchToSpace, 256, 50, 50, 1, 2, 0, 0, 0, 0);
85 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/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/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/cc/gradients/ |
| D | array_grad_test.cc | 299 auto y = BatchToSpace(scope_, x, paddings, /* block_size */ 2); in TEST_F()
|
| D | array_grad.cc | 319 BatchToSpace(scope, grad_inputs[0], op.input(1), block_size)); in SpaceToBatchGrad()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/g3doc/ |
| D | cpu_supported_ops.md | 32 `BatchToSpace` | `Tidx={int32,int64}`<br>`T={bool,complex64,double,float,int…
|
| D | gpu_supported_ops.md | 32 `BatchToSpace` | `Tidx={int32,int64}`<br>`T={bool,complex64,double,float,int…
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | randomized_tests.cc | 1228 TEST_F(OpTest, BatchToSpace) { in TEST_F() argument
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.raw_ops.pbtxt | 376 name: "BatchToSpace"
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.raw_ops.pbtxt | 376 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"
|
| D | ops_history.v1.pbtxt | 10346 name: "BatchToSpace"
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 23868 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 | 3872 name: "BatchToSpace"
|