Searched refs:block_shape_ (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | batch_to_space_nd_test.cc | 35 PopulateTensor<int>(block_shape_, data); in SetBlockShape()
50 int block_shape_; member in tflite::__anon9f6216280111::BatchToSpaceNDOpModel
68 block_shape_ = AddConstInput(TensorType_INT32, block_shape, {2}); in BatchToSpaceNDOpConstModel()
92 block_shape_ = AddInput(TensorType_INT32); in BatchToSpaceNDOpDynamicModel()
|
| D | space_to_batch_nd_test.cc | 40 PopulateTensor<int>(block_shape_, data); in SetBlockShape()
58 int block_shape_; member in tflite::__anonf57345680111::SpaceToBatchNDOpModel
76 block_shape_ = AddConstInput(TensorType_INT32, block_shape, {2}); in SpaceToBatchNDOpConstModel()
100 block_shape_ = AddInput(TensorType_INT32); in SpaceToBatchNDOpDynamicModel()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | spacetobatch_op.cc | 233 block_shape_ = Tensor(tensorflow::DT_INT64, TensorShape({2})); in SpaceToBatchOp()
234 auto block_shape_vec = block_shape_.vec<int64>(); in SpaceToBatchOp()
249 context, in0, block_shape_, in1)); in Compute()
254 Tensor block_shape_; member in tensorflow::SpaceToBatchOp
|
| D | batchtospace_op.cc | 228 block_shape_ = Tensor(tensorflow::DT_INT64, TensorShape({2})); in BatchToSpaceOp()
229 auto block_shape_vec = block_shape_.vec<int64>(); in BatchToSpaceOp()
245 BatchToSpaceOpCompute<Device, T>(context, in0, block_shape_, in1); in Compute()
250 Tensor block_shape_; member in tensorflow::BatchToSpaceOp
|
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | nnapi_delegate_test.cc | 1600 PopulateTensor<int>(block_shape_, data); in SetBlockShape()
1612 int block_shape_; member in tflite::__anon0c68fcb70111::SpaceToBatchNDOpModel
1623 block_shape_ = AddConstInput(TensorType_INT32, block_shape, {2}); in SpaceToBatchNDOpConstModel()
|