- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for input_size (0.19 sec)
-
tensorflow/compiler/mlir/lite/transforms/lower_static_tensor_list.cc
Location loc = op.getLoc(); Value scalar_zero = CreateI32SplatConst(loc, &rewriter, {}, 0); // Compute the input tensorlist's length and store it in `input_size`. IntegerType shape_dtype = rewriter.getIntegerType(32); auto input_size = rewriter.create<TF::TensorListLengthOp>( loc, tensorflow::GetTypeFromTFTensorShape({}, shape_dtype), op.getOperand(0));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 20:00:43 UTC 2024 - 70.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/prepare_tf.cc
mlir::dyn_cast<RankedTensorType>(input.getType()); if (!input_type) { return failure(); } const ArrayRef<int64_t> input_shape = input_type.getShape(); const int input_size = input_shape.size(); RankedTensorType begin_type = mlir::cast<RankedTensorType>(begin.getType()); const ArrayRef<int64_t> begin_shape = begin_type.getShape(); const int begin_dim = begin_shape.size();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 21:49:50 UTC 2024 - 64.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_ops.td
TF_Float32Tensor:$sorted_gains, TF_Float32Tensor:$embedding_table, TF_Int32Tensor:$num_minibatches_per_physical_sparse_core, ConfinedAttr<I64Attr, [IntMinValue<0>]>:$input_size, OptionalAttr<F32Attr>:$quantization_config_low, OptionalAttr<F32Attr>:$quantization_config_high, OptionalAttr<I64Attr>:$quantization_config_num_buckets, StrAttr:$table_name );
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 24 04:08:35 UTC 2024 - 90.5K bytes - Viewed (0)