Home
last modified time | relevance | path

Searched refs:get_item (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/python/autograph/operators/
Dslices_test.py43 t = slices.get_item(
51 t = slices.get_item(initial_str, 1,
58 t = slices.get_item(initial_list_str, 1,
Dslices.py38 def get_item(target, i, opts): function
D__init__.py61 from tensorflow.python.autograph.operators.slices import get_item
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dtensor_list_ops_decomposition.cc688 TF::TensorListGetItemOp get_item, in HandleTensorListGetItemOp() argument
690 auto buffer = get_item.input_handle(); in HandleTensorListGetItemOp()
693 get_item.emitOpError("found tf.TensorListGetItemOp on unknown TensorList."); in HandleTensorListGetItemOp()
696 OpBuilder builder(get_item); in HandleTensorListGetItemOp()
697 auto index = cutil::ReshapeScalarToSizeType(builder, get_item.index(), in HandleTensorListGetItemOp()
698 get_item.getLoc()); in HandleTensorListGetItemOp()
700 cutil::GetElement(index, buffer, OpBuilder(get_item), get_item.getLoc()); in HandleTensorListGetItemOp()
701 get_item.item().replaceAllUsesWith(element); in HandleTensorListGetItemOp()
702 get_item.erase(); in HandleTensorListGetItemOp()
844 } else if (auto get_item = llvm::dyn_cast<TF::TensorListGetItemOp>(&op)) { in DecomposeTensorListOpsInternal() local
[all …]
/external/mesa3d/src/compiler/nir/
Dnir_algebraic.py894 def get_item(opcode, children, pattern=None): function
905 self.wildcard = get_item("__wildcard", ())
906 self.const = get_item("__const", ())
935 item = get_item(opcode, children, pattern)
/external/python/cpython3/Lib/test/
Dtest_buffer.py460 def get_item(lst, indices): function
835 item2 = get_item(lst, ind)