Searched refs:dims_type (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | fill_test.cc | 38 template <typename dims_type, typename value_type>
43 std::initializer_list<dims_type> dims_data, in FillOpModel()
59 PopulateTensor<dims_type>(dims_, dims_data); in FillOpModel()
76 template <typename dims_type, typename quant_type>
81 std::initializer_list<dims_type> dims_data, in QuantizedFillOpModel()
91 PopulateTensor<dims_type>(dims_, dims_data); in QuantizedFillOpModel()
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | unfuse_batch_norm.cc | 41 auto dims_type = RankedTensorType::get({1}, b.getIntegerType(64)); in BroadcastToFeatureDim() local
42 auto dims = DenseIntElementsAttr::get(dims_type, {feature_dim}); in BroadcastToFeatureDim()
94 auto dims_type = RankedTensorType::get({0}, b.getIntegerType(64)); in MaterializeEpsilon() local
95 auto dims = DenseIntElementsAttr::get(dims_type, SmallVector<int64_t, 1>{}); in MaterializeEpsilon()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops_helpers.inc | 524 auto dims_type = dims.getType().dyn_cast<RankedTensorType>();
525 if (!dims_type) return success();
526 if (dims_type.getRank() > 1)
|