Home
last modified time | relevance | path

Searched refs:SpaceToBatch (Results 1 – 20 of 20) sorted by relevance

/external/tensorflow/tensorflow/core/api_def/java_api/
Dapi_def_SpaceToBatch.pbtxt2 graph_op_name: "SpaceToBatch"
4 name: "nn.SpaceToBatch"
/external/tensorflow/tensorflow/core/kernels/
Dspacetobatch_benchmark_test.cc79 BM_SpaceToBatch(SpaceToBatch, 64, 100, 100, 64, 2, 0, 0, 0, 0);
80 BM_SpaceToBatch(SpaceToBatch, 64, 100, 100, 1, 2, 0, 0, 0, 0);
81 BM_SpaceToBatch(SpaceToBatch, 64, 100, 100, 64, 2, 3, 3, 3, 3);
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dspacetobatch_op.cc24 void SpaceToBatch(XlaOpKernelContext* ctx, const xla::XlaOp& input, in SpaceToBatch() function
159 SpaceToBatch(ctx, ctx->Input(0), input_type(0), ctx->InputShape(0), in Compile()
181 SpaceToBatch(ctx, ctx->Input(0), input_type(0), ctx->InputShape(0), in Compile()
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_SpaceToBatch.pbtxt2 graph_op_name: "SpaceToBatch"
99 summary: "SpaceToBatch for 4-D tensors of type T."
Dapi_def_BatchToSpace.pbtxt99 cropping. This is the reverse transformation of SpaceToBatch. More specifically,
Dapi_def_BatchToSpaceND.pbtxt137 reverse of SpaceToBatch. See below for a precise description.
Dapi_def_SpaceToBatchND.pbtxt129 summary: "SpaceToBatch for N-D tensors of type T."
/external/tensorflow/tensorflow/core/api_def/python_api/
Dapi_def_SpaceToBatch.pbtxt2 graph_op_name: "SpaceToBatch"
/external/tensorflow/tensorflow/cc/gradients/
Darray_grad_test.cc280 auto y = SpaceToBatch(scope_, x, paddings, /* block_size */ 2); in TEST_F()
Darray_grad.cc343 SpaceToBatch(scope, grad_inputs[0], op.input(1), block_size)); in BatchToSpaceGrad()
/external/tensorflow/tensorflow/compiler/tf2xla/g3doc/
Dgpu_supported_ops.md218 `SpaceToBatch` | `Tpaddings={int32,int64}`<br>`T={bool,complex64,double,floa…
Dcpu_supported_ops.md221 `SpaceToBatch` | `Tpaddings={int32,int64}`<br>`T={bool,complex64,double,floa…
/external/tensorflow/tensorflow/compiler/tests/
Drandomized_tests.cc2955 TEST_F(OpTest, SpaceToBatch) { in TEST_F() argument
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.raw_ops.pbtxt3396 name: "SpaceToBatch"
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.raw_ops.pbtxt3396 name: "SpaceToBatch"
/external/tensorflow/tensorflow/core/ops/compat/
Dops_history.v0.pbtxt31703 name: "SpaceToBatch"
31728 name: "SpaceToBatch"
Dops_history.v1.pbtxt71238 name: "SpaceToBatch"
Dops_history.v2.pbtxt70611 name: "SpaceToBatch"
/external/tensorflow/tensorflow/go/op/
Dwrappers.go869 func SpaceToBatch(scope *Scope, input tf.Output, paddings tf.Output, block_size int64) (output tf.O… func
/external/tensorflow/tensorflow/core/ops/
Dops.pbtxt34945 name: "SpaceToBatch"