- Sort Score
- Result 10 results
- Languages All
Results 1 - 5 of 5 for shape_of (0.33 sec)
-
tensorflow/compiler/mlir/quantization/stablehlo/tests/passes/lift_quantizable_spots_as_functions.mlir
// CHECK: %[[CONV:.*]] = stablehlo.convolution(%arg0, %arg1) // CHECK: %[[SHAPE_OF_0:.*]] = shape.shape_of %[[CONV]] // CHECK: %[[DYNAMIC_BROADCAST_IN_DIM_0:.*]] = stablehlo.dynamic_broadcast_in_dim %arg2, %[[SHAPE_OF_0]] // CHECK: %[[ADD:.*]] = stablehlo.add %[[CONV]], %[[DYNAMIC_BROADCAST_IN_DIM_0]] // CHECK: %[[SHAPE_OF_1:.*]] = shape.shape_of %[[ADD]] // CHECK-DAG: %[[CONST:.*]] = stablehlo.constant dense<0.000000e+00>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 10 04:07:09 UTC 2024 - 49.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/flatbuffer_import.cc
mlir::DenseIntElementsAttr shape_attr; if (matchPattern(op_state.operands[1], m_Constant(&shape_attr))) { auto shape_ty = op_state.operands[1].getType().dyn_cast<RankedTensorType>(); if (shape_ty != nullptr && shape_ty.hasRank() && shape_ty.getRank() > 1) { llvm::SmallVector<mlir::Attribute, 4> shape; int32_t dim_size = 0; for (const auto& dim :
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 21 18:21:50 UTC 2024 - 66.8K bytes - Viewed (0) -
tensorflow/c/eager/c_api_test.cc
ASSERT_TRUE(TF_GetCode(status.get()) == TF_OK) << TF_Message(status.get()); TFE_Op* shape_op = ShapeOp(ctx, hgpu); TFE_OpSetDevice(shape_op, gpu_device_name.c_str(), status.get()); ASSERT_TRUE(TF_GetCode(status.get()) == TF_OK) << TF_Message(status.get()); TFE_TensorHandle* retvals[1]; int num_retvals = 1; TFE_Execute(shape_op, &retvals[0], &num_retvals, status.get());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Aug 03 20:50:20 UTC 2023 - 94.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/passes/extract_outside_compilation.cc
std::string serialized_func_module; if (HasDynamicOutputs(external_outputs)) { FuncOp shape_op = BuildFunction(clustered_ops, external_operands, external_outputs, &builder); EncapsulateFuncAndSerialize(shape_op, &serialized_func_module); } builder.setInsertionPoint(&op); SmallVector<Value, 4> host_compute_outputs;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 21:25:12 UTC 2024 - 68.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/lower_tf.cc
loc, GetScalarOfType( builder->getIntegerType(use_32bit.getValue() ? 32 : 64), ranked_ty.getDimSize(idx))); } } auto shape = builder->create<TF::ShapeOp>(loc, input, use_32bit); return builder->create<TF::StridedSliceOp>( loc, mlir::RankedTensorType::get({}, getElementTypeOrSelf(shape)), shape, /*begin=*/
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 74.9K bytes - Viewed (0)