Home
last modified time | relevance | path

Searched refs:SpaceToBatch (Results 1 – 24 of 24) 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.cc84 BM_SpaceToBatch(SpaceToBatch, 64, 100, 100, 64, 2, 0, 0, 0, 0);
85 BM_SpaceToBatch(SpaceToBatch, 64, 100, 100, 1, 2, 0, 0, 0, 0);
86 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/python_api/
Dapi_def_SpaceToBatch.pbtxt2 graph_op_name: "SpaceToBatch"
/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/ops/compat/ops_history_v1/
DSpaceToBatch.pbtxt2 name: "SpaceToBatch"
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/
DSpaceToBatch.pbtxt2 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.cc352 SpaceToBatch(scope, grad_inputs[0], op.input(1), block_size)); in BatchToSpaceGrad()
/external/tensorflow/tensorflow/lite/g3doc/guide/
Dop_select_allowlist.md492 * `raw_ops.SpaceToBatch`
/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.cc2956 TEST_F(OpTest, SpaceToBatch) { in TEST_F() argument
/external/tensorflow/tensorflow/compiler/mlir/tosa/g3doc/
Dlegalization.md2749 SpaceToBatch for N-D tensors of type T.
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_generated_ops.td987 cropping. This is the reverse transformation of SpaceToBatch. More specifically,
1096 reverse of SpaceToBatch. See below for a precise description.
13850 def TF_SpaceToBatchOp : TF_Op<"SpaceToBatch", [NoSideEffect]> {
13851 let summary = "SpaceToBatch for 4-D tensors of type T.";
13963 let summary = "SpaceToBatch for N-D tensors of type T.";
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.raw_ops.pbtxt4188 name: "SpaceToBatch"
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.raw_ops.pbtxt4188 name: "SpaceToBatch"
/external/tensorflow/tensorflow/core/ops/compat/
Dops_history.v0.pbtxt31703 name: "SpaceToBatch"
31728 name: "SpaceToBatch"
Dops_history.v2.pbtxt69948 name: "SpaceToBatch"
/external/tensorflow/tensorflow/go/op/
Dwrappers.go1451 func SpaceToBatch(scope *Scope, input tf.Output, paddings tf.Output, block_size int64) (output tf.O… func
/external/tensorflow/tensorflow/core/ops/
Dops.pbtxt46857 name: "SpaceToBatch"