Searched refs:s_param (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | dense_to_sparse.cc | 219 OpBuilder* builder, SparsityParameterAttr* s_param) { in BuildSparsityParameterAttribute() argument
265 *s_param = SparsityParameterAttr::get( in BuildSparsityParameterAttribute()
341 SparsityParameterAttr s_param; in runOnFunction() local
353 &builder, &s_param); in runOnFunction()
360 op->getLoc(), cst.value(), s_param, new_value); in runOnFunction()
371 &s_param); in runOnFunction()
382 op->getLoc(), cst.value(), s_param, new_value); in runOnFunction()
396 &builder, &s_param); in runOnFunction()
403 op->getLoc(), cst.qtypeAttr(), cst.value(), s_param, new_value); in runOnFunction()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | test_util.h | 256 flatbuffers::Offset<SparsityParameters> s_param = CreateSparsityParameters( in AddConstSparseInput() local
280 /*name=*/0, /*quantization=*/0, /*is_variable=*/false, s_param)); in AddConstSparseInput()
328 flatbuffers::Offset<SparsityParameters> s_param = CreateSparsityParameters( variable
368 /*name=*/0, q_params, /*is_variable=*/false, s_param));
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ir/ |
| D | tfl_ops.td | 834 SparsityParameterAttr:$s_param,
840 OpBuilderDAG<(ins "Attribute":$value, "SparsityParameterAttr":$s_param,
845 $_state.addAttribute("s_param", s_param);
3633 SparsityParameterAttr:$s_param,
3641 "SparsityParameterAttr":$s_param, "Attribute":$compressed_data),
3646 $_state.addAttribute("s_param", s_param);
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ |
| D | flatbuffer_export.cc | 791 s_params = BuildSparsityParameters(cst.s_param()); in BuildTensor()
793 s_params = BuildSparsityParameters(cst.s_param()); in BuildTensor()
|